add todo comments

gretelai · Jun 14, 2023 · 393800a · 393800a
1 parent f50958b
commit 393800a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/gretel_trainer/relational/strategies/ancestral.py b/src/gretel_trainer/relational/strategies/ancestral.py
@@ -19,6 +19,7 @@ class AncestralStrategy:
     def name(self) -> str:
         return "ancestral"
 
+    # TODO: remove when `gretel_model` param is removed
     @property
     def supported_gretel_models(self) -> list[str]:
         return ["amplify"]

diff --git a/src/gretel_trainer/relational/strategies/independent.py b/src/gretel_trainer/relational/strategies/independent.py
@@ -20,6 +20,7 @@ class IndependentStrategy:
     def name(self) -> str:
         return "independent"
 
+    # TODO: remove when `gretel_model` param is removed
     @property
     def supported_gretel_models(self) -> list[str]:
         return ["amplify", "actgan", "lstm", "tabular-dp"]