Add ACTGAN to models supported by Ancestral strategy

gretelai · Sep 11, 2023 · 0ccd007 · 0ccd007
1 parent 5b03a30
commit 0ccd007
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gretel_trainer/relational/strategies/ancestral.py b/src/gretel_trainer/relational/strategies/ancestral.py
@@ -24,11 +24,11 @@ def name(self) -> str:
     # TODO: remove when `gretel_model` param is removed
     @property
     def supported_gretel_models(self) -> list[str]:
-        return ["amplify"]
+        return ["amplify", "actgan"]
 
     @property
     def supported_model_keys(self) -> list[str]:
-        return ["amplify"]
+        return ["amplify", "actgan"]
 
     def label_encode_keys(
         self, rel_data: RelationalData, tables: dict[str, pd.DataFrame]