Add get_auto_ablate_params

howsoai · Jan 22, 2024 · fff8f92 · fff8f92
1 parent d0abd0c
commit fff8f92
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/howso/direct/client.py b/howso/direct/client.py
@@ -5075,6 +5075,13 @@ def set_auto_analyze_params(  # noqa: C901
         )
         self._auto_persist_trainee(trainee_id)
 
+    def get_auto_ablate_params(self, trainee_id: str):
+        """
+        Get parameters set by :meth:`set_auto_ablate_params`.
+        """
+        self._auto_resolve_trainee(trainee_id)
+        self.howso.get_auto_ablate_params(trainee_id)
+
     def set_auto_ablate_params(
         self,
         trainee_id: str,

diff --git a/howso/direct/core.py b/howso/direct/core.py
@@ -756,6 +756,14 @@ def get_num_training_cases(self, trainee_id: str) -> Dict:
             A dictionary containing the key "count".
         """
         return self._execute("get_num_training_cases", {"trainee": trainee_id})
+
+    def get_auto_ablate_params(self, trainee_id: str):
+        """
+        Get parameters set by :meth:`set_auto_ablate_params`.
+        """
+        return self._execute(
+            "set_auto_ablate_params", {"trainee": trainee_id}
+        )
 
     def set_auto_ablate_params(
         self,

diff --git a/howso/engine/trainee.py b/howso/engine/trainee.py
@@ -963,6 +963,15 @@ def auto_analyze(self) -> None:
         else:
             raise ValueError("Client must have the 'auto_analyze' method.")
 
+    def get_auto_ablate_params(self):
+        """
+        Get parameters set by :meth:`set_auto_ablate_params`.
+        """
+        if isinstance(self.client, AbstractHowsoClient):
+            self.client.get_auto_ablate_params()
+        else:
+            raise ValueError("Client must have the 'get_auto_ablate_params' method.")
+
     def set_auto_ablate_params(
         self,
         auto_ablate_enabled: bool = False,