increase timeout

srinivasupadhya · Nov 8, 2021 · 0914b05 · 0914b05
1 parent 87b6d5c
commit 0914b05
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/core/test_policies.py b/tests/core/test_policies.py
@@ -173,6 +173,7 @@ def test_featurizer(
         assert loaded.featurizer.max_history == self.max_history
         assert isinstance(loaded.featurizer.state_featurizer, SingleStateFeaturizer)
 
+    @pytest.mark.timeout(120, func_only=True)
     @pytest.mark.parametrize("should_finetune", [False, True])
     def test_persist_and_load(
         self,