Doc updates

CrayLabs · Oct 2, 2023 · e83fdd8 · e83fdd8
1 parent f9a02e1
commit e83fdd8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/doc/data_structures.rst b/doc/data_structures.rst
@@ -358,6 +358,7 @@ are uniform across all SmartRedis clients, and as an example, the C++
                    const std::string& device,
                    int batch_size = 0,
                    int min_batch_size = 0,
+                   int min_batch_timeout = 0,
                    const std::string& tag = "",
                    const std::vector<std::string>& inputs
                        = std::vector<std::string>(),

diff --git a/tests/python/test_errors.py b/tests/python/test_errors.py
@@ -444,6 +444,8 @@ def test_bad_type_set_model_from_file_multigpu(use_cluster, context):
         c.set_model_from_file_multigpu("simple_cnn", modelfile, "TORCH", 0, 1, batch_size="not_an_integer")
     with pytest.raises(TypeError):
         c.set_model_from_file_multigpu("simple_cnn", modelfile, "TORCH", 0, 1, min_batch_size="not_an_integer")
+    with pytest.raises(TypeError):
+        c.set_model_from_file_multigpu("simple_cnn", modelfile, "TORCH", 0, 1, min_batch_timeout="not_an_integer")
     with pytest.raises(TypeError):
         c.set_model_from_file_multigpu("simple_cnn", modelfile, "TORCH", 0, 1, tag=42)