Experiments with CLI cmd passing

Added temporary debug print statements
Lightning-AI · Nov 9, 2024 · ecfa137 · ecfa137
1 parent 0dd0024
commit ecfa137
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/lightning/pytorch/cli.py b/src/lightning/pytorch/cli.py
@@ -138,7 +138,10 @@ def add_lightning_class_args(
             if issubclass(lightning_class, Callback):
                 self.callback_keys.append(nested_key)
             if subclass_mode:
+                print("Subclass mode")
+                print(nested_key)
                 return self.add_subclass_arguments(lightning_class, nested_key, fail_untyped=False, required=required)
+            print(nested_key)
             return self.add_class_arguments(
                 lightning_class,
                 nested_key,