update osiris version

gizatechxyz · Jan 22, 2024 · d1bd0b3 · d1bd0b3
1 parent 2886e99
commit d1bd0b3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,7 +18,7 @@ pyyaml = "^6.0.1"
 prefect-docker = "^0.4.1"
 giza-cli = "^0.6.0"
 distlib = "^0.3.8"
-giza-osiris = "0.1.8"
+giza-osiris = "0.2.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^6.2.5"

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -12,6 +12,6 @@ def test_predict_success():
     arr = np.array([[1, 2], [3, 4]], dtype=np.uint32)
 
     result = model.predict(
-        input_feed={"arr_1": arr}, verifiable=True, output_dtype='tensor_uint')
+        input_feed={"arr_1": arr}, verifiable=True, output_dtype='tensor_int')
 
     assert np.array_equal(result, arr)