output type hints

capitalone · Nov 15, 2023 · 8bbde7d · 8bbde7d
1 parent dd846f0
commit 8bbde7d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rubicon_ml/client/experiment.py b/rubicon_ml/client/experiment.py
@@ -404,7 +404,7 @@ def _get_experiments_from_tags(self, tag_key: str):
 
         return experiments
 
-    def get_child_experiments(self):
+    def get_child_experiments(self) -> List[Experiment]:
         """Get the experiments that are tagged as children of this experiment.
 
         Returns
@@ -414,7 +414,7 @@ def get_child_experiments(self):
         """
         return self._get_experiments_from_tags("child")
 
-    def get_parent_experiments(self):
+    def get_parent_experiments(self) -> List[Experiment]:
         """Get the experiments that are tagged as parents of this experiment.
 
         Returns