diff --git a/clarifai/runners/models/model_run_locally.py b/clarifai/runners/models/model_run_locally.py index 6164a4bf..d83fa326 100644 --- a/clarifai/runners/models/model_run_locally.py +++ b/clarifai/runners/models/model_run_locally.py @@ -224,6 +224,7 @@ def _run_test(self): nodepool_id="n/a", compute_cluster_id="n/a", user_id="n/a", + health_check_port=None, ) # send an inference. self._run_model_inference(runner) diff --git a/requirements.txt b/requirements.txt index 6fb2f9b2..dec59390 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ -clarifai-grpc>=11.0.4 -clarifai-protocol>=0.0.14 +clarifai-grpc>=11.0.5 +clarifai-protocol>=0.0.16 numpy>=1.22.0 tqdm>=4.65.0 rich>=13.4.2 diff --git a/tests/runners/hf_mbart_model/requirements.txt b/tests/runners/hf_mbart_model/requirements.txt index fdb027d6..63d7193f 100644 --- a/tests/runners/hf_mbart_model/requirements.txt +++ b/tests/runners/hf_mbart_model/requirements.txt @@ -4,4 +4,4 @@ transformers>=4.44 accelerate>=1.0.1 optimum>=1.20.0 sentencepiece==0.2.0 -requests==2.23.0 +requests diff --git a/tests/runners/test_runners.py b/tests/runners/test_runners.py index ceb2a90e..a9aae234 100644 --- a/tests/runners/test_runners.py +++ b/tests/runners/test_runners.py @@ -183,6 +183,7 @@ def setup_class(cls): num_parallel_polls=1, base_url=cls.AUTH.base, user_id=cls.AUTH.user_id, + health_check_port=None, ) cls.thread = threading.Thread(target=cls.runner.start) cls.thread.daemon = True # close when python closes @@ -499,6 +500,7 @@ def setup_class(cls): num_parallel_polls=1, base_url=cls.AUTH.base, user_id=cls.AUTH.user_id, + health_check_port=None, ) cls.model = Model( user_id=cls.AUTH.user_id,