diff --git a/tango/integrations/wandb/step_cache.py b/tango/integrations/wandb/step_cache.py index 7195486b8..56ea8b84c 100644 --- a/tango/integrations/wandb/step_cache.py +++ b/tango/integrations/wandb/step_cache.py @@ -142,7 +142,7 @@ def use_step_result_artifact(self, step: Union[Step, StepInfo]) -> None: def _download_step_remote(self, step_result, target_dir: PathOrStr): try: - step_result.download(root=target_dir, recursive=True) + step_result.download(root=target_dir) except (WandbError, ValueError): raise RemoteNotFoundError()