New plugin: AIM (#216)

* Initial aim plugin

* Fixing linting

* Add tests and fix bandit

* Fix test names

* Fix tests

* Create data folder for categorical data

* Tests passing and tutorial

* updated tutorial

* Fix tests

* Move AIM to privacy from generic

.pre-commit-config.yaml

@@ -1,52 +1,51 @@
-exclude: '^docs/conf.py'
+exclude: "^docs/conf.py"
 
 repos:
-- repo: https://github.com/pre-commit/pre-commit-hooks
-  rev: v4.0.1
-  hooks:
-  - id: trailing-whitespace
-  - id: check-ast
-  - id: check-json
-  - id: check-merge-conflict
-  - id: check-xml
-  - id: check-yaml
-  - id: debug-statements
-  - id: end-of-file-fixer
-  - id: requirements-txt-fixer
-  - id: mixed-line-ending
-    args: ['--fix=auto']  # replace 'auto' with 'lf' to enforce Linux/Mac line endings or 'crlf' for Windows
-- repo: https://github.com/psf/black
-  rev: 22.3.0
-  hooks:
-  - id: black
-    language_version: python3
-- repo: https://github.com/pycqa/isort
-  rev: 5.12.0
-  hooks:
-  - id: isort
-- repo: https://github.com/PyCQA/flake8
-  rev: 6.0.0
-  hooks:
-  - id: flake8
-    args: [
-        "--max-line-length=480",
-        "--extend-ignore=E203,W503"
-    ]
-- repo: https://github.com/pre-commit/mirrors-mypy
-  rev: v0.991
-  hooks:
-  - id: mypy
-    args: [
-          "--ignore-missing-imports",
-          "--scripts-are-modules",
-          "--disallow-incomplete-defs",
-          "--no-implicit-optional",
-          "--warn-unused-ignores",
-          "--warn-redundant-casts",
-          "--strict-equality",
-          "--warn-unreachable",
-          "--disallow-untyped-defs",
-          "--disallow-untyped-calls",
-          "--install-types",
-          "--non-interactive",
-      ]
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.0.1
+    hooks:
+      - id: trailing-whitespace
+      - id: check-ast
+      - id: check-json
+      - id: check-merge-conflict
+      - id: check-xml
+      - id: check-yaml
+      - id: debug-statements
+      - id: end-of-file-fixer
+      - id: requirements-txt-fixer
+      - id: mixed-line-ending
+        args: ["--fix=auto"] # replace 'auto' with 'lf' to enforce Linux/Mac line endings or 'crlf' for Windows
+  - repo: https://github.com/psf/black
+    rev: 22.3.0
+    hooks:
+      - id: black
+        language_version: python3
+  - repo: https://github.com/pycqa/isort
+    rev: 5.12.0
+    hooks:
+      - id: isort
+  - repo: https://github.com/PyCQA/flake8
+    rev: 6.0.0
+    hooks:
+      - id: flake8
+        args: ["--max-line-length=480", "--extend-ignore=E203,W503"]
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v0.991
+    hooks:
+      - id: mypy
+        args: [
+            "--ignore-missing-imports",
+            "--scripts-are-modules",
+            "--disallow-incomplete-defs",
+            "--no-implicit-optional",
+            "--warn-unused-ignores",
+            "--warn-redundant-casts",
+            "--strict-equality",
+            "--warn-unreachable",
+            "--disallow-untyped-defs",
+            "--disallow-untyped-calls",
+            "--install-types",
+            "--non-interactive",
+            "--follow-imports=skip", # This is temporary until the mbi directory is not excluded
+          ]
+        exclude: ^src/synthcity/plugins/core/models/mbi/ # This is temporary until the mbi directory is fully typed

README.md

@@ -5,7 +5,7 @@
 </h2>
 
 <h4 align="center">
-    A library for generating and evaluating synthetic tabular data..
+    A library for generating and evaluating synthetic tabular data.
 </h4>
 
 
@@ -20,7 +20,7 @@
 
 [![](https://pepy.tech/badge/synthcity)](https://pypi.org/project/synthcity/)
 [![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://github.com/vanderschaarlab/synthcity/blob/main/LICENSE)
-[![Python 3.7+](https://img.shields.io/badge/python-3.7+-blue.svg)](https://www.python.org/downloads/release/python-370/)
+[![Python 3.8+](https://img.shields.io/badge/python-3.8+-blue.svg)](https://www.python.org/downloads/release/python-380/)
 [![about](https://img.shields.io/badge/about-The%20van%20der%20Schaar%20Lab-blue)](https://www.vanderschaar-lab.com/)
 [![slack](https://img.shields.io/badge/chat-on%20slack-purple?logo=slack)](https://join.slack.com/t/vanderschaarlab/shared_invite/zt-1pzy8z7ti-zVsUPHAKTgCd1UoY8XtTEw)
 

docs/examples.rst

@@ -26,6 +26,9 @@ General-purpose generators
     CTGAN <tutorials/plugins/generic/plugin_ctgan.ipynb>
     Normalizing Flows <tutorials/plugins/generic/plugin_nflow.ipynb>
     TVAE <tutorials/plugins/generic/plugin_tvae.ipynb>
+    GOGGLE <generated/synthcity.plugins.generic.plugin_goggle.rst>
+    ARF <generated/synthcity.plugins.generic.plugin_arf.rst>
+    GReaT <generated/synthcity.plugins.generic.plugin_great.rst>
 
 
 Time-series generators
@@ -49,6 +52,7 @@ Privacy-related generators
     AdsGAN <tutorials/plugins/privacy/plugin_adsgan.ipynb>
     PATEGAN <tutorials/plugins/privacy/plugin_pategan.ipynb>
     PrivBayes <tutorials/plugins/privacy/plugin_privbayes.ipynb>
+    AIM <generated/synthcity.plugins.generic.plugin_aim.rst>
 
 Domain adaptation generators
 ------------------------------

docs/generators.rst

@@ -12,6 +12,9 @@ General purpose
     Normalizing Flows <generated/synthcity.plugins.generic.plugin_nflow.rst>
     RTVAE <generated/synthcity.plugins.generic.plugin_rtvae.rst>
     TVAE <generated/synthcity.plugins.generic.plugin_tvae.rst>
+    GOGGLE <generated/synthcity.plugins.generic.plugin_goggle.rst>
+    ARF <generated/synthcity.plugins.generic.plugin_arf.rst>
+    GReaT <generated/synthcity.plugins.generic.plugin_great.rst>
 
 Privacy-focused
 -----------------
@@ -24,6 +27,7 @@ Privacy-focused
     PrivBayes <generated/synthcity.plugins.privacy.plugin_privbayes.rst>
     DP-GAN <generated/synthcity.plugins.privacy.plugin_dpgan.rst>
     DECAF <generated/synthcity.plugins.privacy.plugin_decaf.rst>
+    AIM <generated/synthcity.plugins.generic.plugin_aim.rst>
 
 Domain adaptation
 -------------------