Merge pull request #868 from basetenlabs/bump-version-0.9.5

Release 0.9.5
basetenlabs · Mar 19, 2024 · 90af43f · 90af43f
2 parents bcc5a05 + 020542b
commit 90af43f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truss"
-version = "0.9.4"
+version = "0.9.5"
 description = "A seamless bridge from model development to model delivery"
 license = "MIT"
 readme = "README.md"

diff --git a/truss/tests/test_config.py b/truss/tests/test_config.py
@@ -87,6 +87,7 @@ def test_parse_resources(input_dict, expect_resources, output_dict):
         ("A10G:4", AcceleratorSpec(Accelerator.A10G, 4)),
         ("A100:8", AcceleratorSpec(Accelerator.A100, 8)),
         ("H100", AcceleratorSpec(Accelerator.H100, 1)),
+        ("H100_40GB", AcceleratorSpec(Accelerator.H100_40GB, 1)),
     ],
 )
 def test_acc_spec_from_str(input_str, expected_acc):

diff --git a/truss/truss_config.py b/truss/truss_config.py
@@ -51,6 +51,7 @@ class Accelerator(Enum):
     V100 = "V100"
     A100 = "A100"
     H100 = "H100"
+    H100_40GB = "H100_40GB"
 
 
 @dataclass