Merge pull request #654 from basetenlabs/bump-version-0.7.4

Release 0.7.4

Author: squidarth

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truss"
-version = "0.7.3"
+version = "0.7.4"
 description = "A seamless bridge from model development to model delivery"
 license = "MIT"
 readme = "README.md"

truss/tests/test_config.py

@@ -1,3 +1,4 @@
+import tempfile
 from pathlib import Path
 
 import pytest
@@ -205,6 +206,14 @@ def test_non_default_train():
     assert new_config == config.to_dict(verbose=False)
 
 
+def test_null_hf_cache_key():
+    config_yaml_dict = {"hf_cache": None}
+    with tempfile.NamedTemporaryFile(mode="w", delete=False) as tmp_file:
+        yaml.safe_dump(config_yaml_dict, tmp_file)
+    config = TrussConfig.from_yaml(Path(tmp_file.name))
+    assert config.hf_cache == HuggingFaceCache.from_list([])
+
+
 def test_huggingface_cache_single_model_default_revision():
     config = TrussConfig(
         python_version="py39",

truss/truss_config.py

@@ -489,7 +489,10 @@ def from_dict(d):
                 d.get("external_data"), ExternalData.from_list
             ),
             base_image=transform_optional(d.get("base_image"), BaseImage.from_dict),
-            hf_cache=HuggingFaceCache.from_list(d.get("hf_cache", [])),
+            hf_cache=transform_optional(
+                d.get("hf_cache") or [],
+                HuggingFaceCache.from_list,
+            ),
         )
         config.validate()
         return config