Update DeepInfraModel_unit_test.py

swarmauri · Sep 25, 2024 · 76df23f · 76df23f
1 parent 4e85072
commit 76df23f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkgs/swarmauri/tests/unit/llms/DeepInfraModel_unit_test.py b/pkgs/swarmauri/tests/unit/llms/DeepInfraModel_unit_test.py
@@ -19,7 +19,7 @@ def deepinfra_model():
 
 @pytest.mark.unit
 def test_ubc_resource(deepinfra_model):
-    assert deepinfra_model.resource == "deepinfra_model"
+    assert deepinfra_model.resource == "LLM"
 
 
 @pytest.mark.unit