Add TODO

ServiceNow · Apr 19, 2023 · 1657e0e · 1657e0e
1 parent 0089720
commit 1657e0e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/azimuth/modules/perturbation_testing/perturbation_testing.py b/azimuth/modules/perturbation_testing/perturbation_testing.py
@@ -59,6 +59,8 @@ class PerturbationTestingModule(DatasetResultModule[PerturbationTestingConfig]):
     """
 
     required_mod_options = {"pipeline_index"}
+    # This module doesn't call self.get_model() but requires the model (predict_task.compute(batch))
+    # TODO Find a more robust way to determine when modules require models.
     worker = Worker.model
 
     def __init__(