diff --git a/tests/policies/multi_task_dit/test_multi_task_dit.py b/tests/policies/multi_task_dit/test_multi_task_dit.py index 88bdd6e24..42fd1a131 100644 --- a/tests/policies/multi_task_dit/test_multi_task_dit.py +++ b/tests/policies/multi_task_dit/test_multi_task_dit.py @@ -16,11 +16,8 @@ """Test script for Multi-Task DiT policy. -To run tests with GPU on Modal (temporary script): - modal run run_tests_modal.py - To run tests locally: - python -m pytest tests/policies/test_multi_task_dit_policy.py -v + python -m pytest tests/policies/multi_task_dit/test_multi_task_dit.py -v """ import pytest