clarifai 11.1.5rc2__py3-none-any.whl → 11.1.5rc3__py3-none-any.whl

clarifai/__init__.py

@@ -1 +1 @@
-__version__ = "11.1.5rc2"
+__version__ = "11.1.5rc3"

clarifai/client/model.py

@@ -443,6 +443,15 @@ class Model(Lister, BaseClient):
         output_config=output_config,
     )
 
+  def predict2(self, inputs):
+    """Predicts the model based on the given inputs.
+
+    Args:
+        inputs (list[Input]): The inputs to predict, must be less than 128.
+    """
+
+    return self.model_client._predict(inputs=inputs,)
+
   def _check_predict_input_type(self, input_type: str) -> None:
     """Checks if the input type is valid for the model.
 

clarifai/client/model_client.py

@@ -48,8 +48,10 @@ class ModelClient:
 
     request = service_pb2.PostModelOutputsRequest()
     request.CopyFrom(self.request_template)
-    request.model.model_version.output_info.params['_method_name'] = '_GET_SIGNATURES'
-    request.inputs.add()  # empty input for this method
+    # request.model.model_version.output_info.params['_method_name'] = '_GET_SIGNATURES'
+    inp = request.inputs.add()  # empty input for this method
+    inp.data.parts.add()  # empty part for this input
+    inp.data.metadata['method_name'] = '_GET_SIGNATURES'
     start_time = time.time()
     backoff_iterator = BackoffIterator(10)
     while True:
@@ -65,8 +67,10 @@ class ModelClient:
       self._method_signatures = {}
       return
     if response.status.code != status_code_pb2.SUCCESS:
-      raise Exception(f"Model failed with response {response.status!r}")
+      raise Exception(f"Model failed with response {response!r}")
     self._method_signatures = signatures_from_json(response.outputs[0].data.string_value)
+    import pdb
+    pdb.set_trace()
 
   def _define_functions(self):
     '''

clarifai/runners/dockerfile_template/Dockerfile.template

@@ -44,8 +44,8 @@ ENV PYTHONPATH=${PYTHONPATH}:/home/nonroot/main \
     CLARIFAI_COMPUTE_CLUSTER_ID=${CLARIFAI_COMPUTE_CLUSTER_ID} \
     CLARIFAI_API_BASE=${CLARIFAI_API_BASE:-https://api.clarifai.com}
 
-# Write out the model function signatures
-RUN ["python", "-m", "clarifai.cli", "model", "signatures", "--model_path", "/home/nonroot/main", "--out_path", "/home/nonroot/main/signatures.yaml"]
+# # Write out the model function signatures
+# RUN ["python", "-m", "clarifai.cli", "model", "signatures", "--model_path", "/home/nonroot/main", "--out_path", "/home/nonroot/main/signatures.yaml"]
 
 # Finally run the clarifai entrypoint to start the runner loop and local dev server.
 # Note(zeiler): we may want to make this a clarifai CLI call.

clarifai/runners/models/model_class.py

@@ -70,7 +70,6 @@ class ModelClass(ABC):
 
   def load_model(self):
     """Load the model."""
-    pass
 
   def _handle_get_signatures_request(self) -> service_pb2.MultiOutputResponse:
     methods = self._get_method_info()
@@ -98,8 +97,9 @@ class ModelClass(ABC):
     outputs = []
     try:
       # TODO add method name field to proto
-      call_params = dict(request.model.model_version.output_info.params)
-      method_name = call_params.get('_method_name', 'predict')
+      method_name = request.inputs[0].data.metadata['method_name']
+      # call_params = dict(request.model.model_version.output_info.params)
+      # method_name = call_params.get('_method_name', 'predict')
       if method_name == '_GET_SIGNATURES':  # special case to fetch signatures, TODO add endpoint for this
         return self._handle_get_signatures_request()
       if method_name not in self._get_method_info():

{clarifai-11.1.5rc2.dist-info → clarifai-11.1.5rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: clarifai
-Version: 11.1.5rc2
+Version: 11.1.5rc3
 Summary: Clarifai Python SDK
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai

{clarifai-11.1.5rc2.dist-info → clarifai-11.1.5rc3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-clarifai/__init__.py,sha256=kO0DOevIoDJN9LwerQz7F5pbvLR_IJKwxl_5djnmxio,26
+clarifai/__init__.py,sha256=cQU5SXsYznf1FLV9LeeLMrjUc1vlOnCbQLM5_Z--QKI,26
 clarifai/cli.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/errors.py,sha256=RwzTajwds51wLD0MVlMC5kcpBnzRpreDLlazPSBZxrg,2605
 clarifai/versions.py,sha256=jctnczzfGk_S3EnVqb2FjRKfSREkNmvNEwAAa_VoKiQ,222
@@ -29,8 +29,8 @@ clarifai/client/dataset.py,sha256=y3zKT_VhP1gyN3OO-b3cPeW21ZXyKbQ7ZJkEG06bsTU,32
 clarifai/client/deployment.py,sha256=w7Y6pA1rYG4KRK1SwusRZc2sQRXlG8wezuVdzSWpCo0,2586
 clarifai/client/input.py,sha256=obMAHMDU1OwfXZ8KraOnGFlWzlW-3F7Ob_2lcOQMlhY,46339
 clarifai/client/lister.py,sha256=03KGMvs5RVyYqxLsSrWhNc34I8kiF1Ph0NeyEwu7nMU,2082
-clarifai/client/model.py,sha256=Cs8EYUBTMfiycNMW1rO8lDEnqos69NPrEAflMBZatH8,75100
-clarifai/client/model_client.py,sha256=5AsB7y59V6i-oChGJqwNByWMdE52snGg5esBIvCgbmc,16506
+clarifai/client/model.py,sha256=Z4cz-qXznbttETnH3W_LyyscGawvtBkRfaSCsdOgAvs,75334
+clarifai/client/model_client.py,sha256=vPdNZbLV8IaTDhyi6dUV_WoPx6we-EFq5gX4MufAAD4,16653
 clarifai/client/module.py,sha256=FTkm8s9m-EaTKN7g9MnLhGJ9eETUfKG7aWZ3o1RshYs,4204
 clarifai/client/nodepool.py,sha256=la3vTFrO4LX8zm2eQ5jqf2L0-kQ63Dano8FibadoZbk,10152
 clarifai/client/search.py,sha256=GaPWN6JmTQGZaCHr6U1yv0zqR6wKFl7i9IVLg2ul1CI,14254
@@ -132,10 +132,10 @@ clarifai/runners/__pycache__/__init__.cpython-310.pyc,sha256=jVTodJyP53yeIEk2sPu
 clarifai/runners/__pycache__/server.cpython-310.pyc,sha256=HfZ6_Vrr5q78nznFqP_aWhPENplHIol_qPaf8lLlotE,3229
 clarifai/runners/dockerfile_template/Dockerfile.debug,sha256=sRlfRmSLE_TiLORcVRx-3-B0vvSNeUYgm0CCrWmLvAA,667
 clarifai/runners/dockerfile_template/Dockerfile.debug~,sha256=7YOVg3adIaiudfSkfLGeyxt-FfIBbD3UIIYccrIVJTs,426
-clarifai/runners/dockerfile_template/Dockerfile.template,sha256=cibinDDE4uRGt2d2ho2T7CduSeDF6R2n7U4_4MAouPw,2521
+clarifai/runners/dockerfile_template/Dockerfile.template,sha256=MWGjj0My6d18v-nyhWuNbSXUzCzR6KWkohRB1yInb7Q,2525
 clarifai/runners/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/runners/models/model_builder.py,sha256=UCAeJiP_fRFfIuVI1sOge7LMwe-psgvdBuZ0LsJQMpM,33062
-clarifai/runners/models/model_class.py,sha256=V7Wf8TiZnccGRJnElr9RU6f8yxzyLXVziEKCfBnLNKU,11060
+clarifai/runners/models/model_class.py,sha256=7pZ8ci4M8Qh91OW-C6yuWAhl3psLDfyvGypaGyFRkLs,11122
 clarifai/runners/models/model_run_locally.py,sha256=VZetm9Mko8MBjcjwr6PCnTU9gF3glgD5qvpbj-8tW2s,17962
 clarifai/runners/models/model_runner.py,sha256=qyc73pe4xc9BsUKHwnOyC9g-RNCARiFis4GTh-yg0vg,6219
 clarifai/runners/models/model_servicer.py,sha256=A--b1P71PBCAMJCpy_-fpNDkfCVdvdMh1LleW15dSas,3037
@@ -195,9 +195,9 @@ clarifai/workflows/__pycache__/__init__.cpython-310.pyc,sha256=oRKg6B7Z-wWQy0EW2
 clarifai/workflows/__pycache__/export.cpython-310.pyc,sha256=cNmGLnww7xVpm4htd1vRhQJoEZ1dhpN1oD8iLLAtVzM,2418
 clarifai/workflows/__pycache__/utils.cpython-310.pyc,sha256=rm2kWk4a3GOKWoerXpEAEeRvGhEe7wPd0ZZ6jHtEGqY,1925
 clarifai/workflows/__pycache__/validate.cpython-310.pyc,sha256=QA1i6YdDpY824cqtQvkEaFPpaCa2iqfOwFouqwZfAKY,2139
-clarifai-11.1.5rc2.dist-info/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
-clarifai-11.1.5rc2.dist-info/METADATA,sha256=nhhZDqrOJR506lzQ3Gi1leNYcT8UuJh3N9mvD1glPQQ,22229
-clarifai-11.1.5rc2.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-clarifai-11.1.5rc2.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
-clarifai-11.1.5rc2.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
-clarifai-11.1.5rc2.dist-info/RECORD,,
+clarifai-11.1.5rc3.dist-info/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
+clarifai-11.1.5rc3.dist-info/METADATA,sha256=Zl3cY8_BDZ33bV5M1VWnPiAMUGT4EIwE3ZqFHWaW0EY,22229
+clarifai-11.1.5rc3.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+clarifai-11.1.5rc3.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
+clarifai-11.1.5rc3.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
+clarifai-11.1.5rc3.dist-info/RECORD,,