setup and readme updated

ThaminduR · Sep 1, 2020 · b8df7c8 · b8df7c8
1 parent 7afb68c
commit b8df7c8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -496,6 +496,7 @@ ________________________________________________________________________________
 * PySpark 2.4.5. You can easily install it with `pip install pyspark`
 * PyArrow `pip install pyarrow`
 * Pandas `pip intall pandas`
+* kmodes `pip install kmodes`
 
 ### Clustering Based K Anonymity
 
@@ -580,7 +581,7 @@ Recommended small number of l to minimum the data loss. Default value is l = 2.
      StructField("column4", StringType()),
  ])
 
-l_df = Anonymizer.l_diverse(k_df,schema, QI, l=2)
+l_df = Anonymizer.l_diverse(k_df,schema, QI,SA, l=2)
 l_df.show()
 ```
 

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="spark_privacy_preserver",  # Replace with your own username
-    version="0.2.1",
+    version="0.3.1",
     author="thamindu",
     author_email="[email protected]",
     description="Anonymizing Library for Apache Spark",
@@ -19,4 +19,16 @@
         "Operating System :: OS Independent",
     ],
     python_requires='>=3.6',
+    install_requires=[
+        'pandas>=1.1',
+        'pyspark==2.4.5',
+        'pyarrow==0.17.1',
+        'diffprivlib==0.2.1',
+        'tabulate==0.8.7',
+        'mypy>=0.770',
+        'kmodes'
+    ],
+    extras_requires={
+        'DPLib': ['notebook']
+    }
 )