From 7bbb87eb259b52d528454a7679eb47329f9fdd67 Mon Sep 17 00:00:00 2001
From: Jiuguang Wang <jw@theaiinstitute.com>
Date: Thu, 20 Jul 2023 11:50:06 -0400
Subject: [PATCH] Fixed CI

---
 test/test_setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/test_setup.py b/test/test_setup.py
index cd4ba24..52f8016 100644
--- 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():