Fixed CI

bdaiinstitute · Jul 20, 2023 · 7bbb87e · 7bbb87e
1 parent 1c30f28
commit 7bbb87e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_setup.py b/test/test_setup.py
@@ -2,6 +2,7 @@
 
 import torch
 from habitat import get_config
+from habitat_baselines.common.baseline_registry import baseline_registry
 
 
 def test_habitat():