Merge pull request #362 from RavirajBaraiya/pip_zingg

python api dir deleted and moved FebrlExample.py
zinggAI · Jun 24, 2022 · 21e134a · 21e134a
2 parents af6b922 + 4bee13f
commit 21e134a
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 398 deletions.
diff --git a/python/examples/FebrlExample.py → examples/febrl/FebrlExample.py b/python/examples/FebrlExample.py → examples/febrl/FebrlExample.py
@@ -1,4 +1,5 @@
-from python.api.zingg.zingg import *
+from zingg import *
+from zingg.pipes import *
 
 #build the arguments for zingg
 args = Arguments()
@@ -28,21 +29,22 @@
 #in that case, replace df with input df
 df = spark.read.format("csv").schema("id string, fname string, lname string, stNo string, add1 string, add2 string, city string, state string, areacode string, dob string, ssn  string").load("examples/febrl/test.csv")
 
-inputPipe = Pipe("test", "csv")
-inputPipe.addProperty("location","examples/febrl/test.csv")
+inputPipe = CsvPipe("test")
+inputPipe.setLocation("examples/febrl/test.csv")
 
 dfSchema = str(df.schema.json())
 inputPipe.setSchema(dfSchema)
 
 args.setData(inputPipe)
 
 #setting outputpipe in 'args'
-outputPipe = Pipe("result", "csv")
-outputPipe.addProperty("location","/tmp")
+outputPipe = CsvPipe("csv")
+outputPipe.setLocation("/tmp")
 
 args.setOutput(outputPipe)
 
-options = ClientOptions(["--phase", "trainMatch",  "--conf", "dummy", "--license", "dummy", "--email", "xxx@yyy.com"])
+options = ClientOptions()
+options.setPhase("trainMatch")
 
 #Zingg execution for the given phase
 zingg = Zingg(args, options)

diff --git a/python/api/__init__.py b/python/api/__init__.py
diff --git a/python/api/setup.cfg b/python/api/setup.cfg
diff --git a/python/api/setup.py b/python/api/setup.py
diff --git a/python/api/zingg/pipes/__init__.py b/python/api/zingg/pipes/__init__.py
diff --git a/python/api/zingg/pipes/pipes.py b/python/api/zingg/pipes/pipes.py