clarifai 11.1.5rc7__tar.gz → 11.1.6__tar.gz

{clarifai-11.1.5rc7/clarifai.egg-info → clarifai-11.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: clarifai
-Version: 11.1.5rc7
+Version: 11.1.6
 Summary: Clarifai Python SDK
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai
@@ -19,8 +19,32 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Provides-Extra: all
 License-File: LICENSE
+Requires-Dist: clarifai-grpc>=11.1.3
+Requires-Dist: clarifai-protocol>=0.0.16
+Requires-Dist: numpy>=1.22.0
+Requires-Dist: tqdm>=4.65.0
+Requires-Dist: rich>=13.4.2
+Requires-Dist: PyYAML>=6.0.1
+Requires-Dist: schema==0.7.5
+Requires-Dist: Pillow>=9.5.0
+Requires-Dist: tabulate>=0.9.0
+Requires-Dist: fsspec>=2024.6.1
+Requires-Dist: click>=8.1.7
+Requires-Dist: requests>=2.32.3
+Provides-Extra: all
+Requires-Dist: pycocotools==2.0.6; extra == "all"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 <h1 align="center">
   <a href="https://www.clarifai.com/"><img alt="Clarifai" title="Clarifai" src="https://github.com/user-attachments/assets/623b883b-7fe5-4b95-bbfa-8691f5779af4"></a>

clarifai-11.1.6/clarifai/__init__.py

@@ -0,0 +1 @@
+__version__ = "11.1.6"

{clarifai-11.1.5rc7 → clarifai-11.1.6}/clarifai/cli/model.py

@@ -9,11 +9,7 @@ def model():
 
 
 @model.command()
-@click.option(
-    '--model_path',
-    type=click.Path(exists=True),
-    required=True,
-    help='Path to the model directory.')
+@click.argument("model_path", type=click.Path(exists=True), required=False, default=".")
 @click.option(
     '--stage',
     required=False,
@@ -30,17 +26,21 @@ def model():
     'Flag to skip generating a dockerfile so that you can manually edit an already created dockerfile.',
 )
 def upload(model_path, stage, skip_dockerfile):
-  """Upload a model to Clarifai."""
+  """Upload a model to Clarifai.
+
+    MODEL_PATH: Path to the model directory. If not specified, the current directory is used by default.
+  """
   from clarifai.runners.models.model_builder import upload_model
   upload_model(model_path, stage, skip_dockerfile)
 
 
 @model.command()
-@click.option(
-    '--model_path',
+@click.argument(
+    "model_path",
     type=click.Path(exists=True),
-    required=True,
-    help='Path to the model directory.')
+    required=False,
+    default=".",
+)
 @click.option(
     '--out_path',
     type=click.Path(exists=False),
@@ -59,7 +59,10 @@ def upload(model_path, stage, skip_dockerfile):
     'The stage we are calling download checkpoints from. Typically this would be in the build stage which is the default. Other options include "runtime" to be used in load_model or "upload" to be used during model upload. Set this stage to whatever you have in config.yaml to force downloading now.'
 )
 def download_checkpoints(model_path, out_path, stage):
-  """Download checkpoints from external source to local model_path"""
+  """Download checkpoints from external source to local model_path
+
+  MODEL_PATH: Path to the model directory. If not specified, the current directory is used by default.
+  """
 
   from clarifai.runners.models.model_builder import ModelBuilder
   builder = ModelBuilder(model_path, download_validation_only=True)
@@ -67,36 +70,12 @@ def download_checkpoints(model_path, out_path, stage):
 
 
 @model.command()
-@click.option(
-    '--model_path',
+@click.argument(
+    "model_path",
     type=click.Path(exists=True),
-    required=True,
-    help='Path to the model directory.')
-@click.option(
-    '--out_path',
-    type=click.Path(exists=False),
     required=False,
-    default=None,
-    help='Path to write the method signature defitions to. If not provided, use stdout.')
-def signatures(model_path, out_path):
-  """Generate method signatures for the model."""
-
-  from clarifai.runners.models.model_builder import ModelBuilder
-  builder = ModelBuilder(model_path, download_validation_only=True)
-  signatures = builder.method_signatures_yaml()
-  if out_path:
-    with open(out_path, 'w') as f:
-      f.write(signatures)
-  else:
-    click.echo(signatures)
-
-
-@model.command()
-@click.option(
-    '--model_path',
-    type=click.Path(exists=True),
-    required=True,
-    help='Path to the model directory.')
+    default=".",
+)
 @click.option(
     '--mode',
     type=click.Choice(['env', 'container'], case_sensitive=False),
@@ -118,7 +97,10 @@ def signatures(model_path, out_path):
     'Keep the Docker image after testing the model locally (applicable for container mode). Defaults to False.'
 )
 def test_locally(model_path, keep_env=False, keep_image=False, mode='env'):
-  """Test model locally."""
+  """Test model locally.
+
+  MODEL_PATH: Path to the model directory. If not specified, the current directory is used by default.
+  """
   try:
     from clarifai.runners.models import model_run_locally
     if mode == 'env' and keep_image:
@@ -139,11 +121,12 @@ def test_locally(model_path, keep_env=False, keep_image=False, mode='env'):
 
 
 @model.command()
-@click.option(
-    '--model_path',
+@click.argument(
+    "model_path",
     type=click.Path(exists=True),
-    required=True,
-    help='Path to the model directory.')
+    required=False,
+    default=".",
+)
 @click.option(
     '--port',
     '-p',
@@ -172,7 +155,10 @@ def test_locally(model_path, keep_env=False, keep_image=False, mode='env'):
     'Keep the Docker image after testing the model locally (applicable for container mode). Defaults to False.'
 )
 def run_locally(model_path, port, mode, keep_env, keep_image):
-  """Run the model locally and start a gRPC server to serve the model."""
+  """Run the model locally and start a gRPC server to serve the model.
+
+  MODEL_PATH: Path to the model directory. If not specified, the current directory is used by default.
+  """
   try:
     from clarifai.runners.models import model_run_locally
     if mode == 'env' and keep_image:
@@ -197,13 +183,17 @@ def run_locally(model_path, port, mode, keep_env, keep_image):
 
 
 @model.command()
-@click.option(
-    '--model_path',
+@click.argument(
+    "model_path",
     type=click.Path(exists=True),
-    required=True,
-    help='Path to the model directory.')
+    required=False,
+    default=".",
+)
 def local_dev(model_path):
-  """Run the model as a local dev runner to help debug your model connected to the API. You must set several envvars such as CLARIFAI_PAT, CLARIFAI_RUNNER_ID, CLARIFAI_NODEPOOL_ID, CLARIFAI_COMPUTE_CLUSTER_ID. """
+  """Run the model as a local dev runner to help debug your model connected to the API. You must set several envvars such as CLARIFAI_PAT, CLARIFAI_RUNNER_ID, CLARIFAI_NODEPOOL_ID, CLARIFAI_COMPUTE_CLUSTER_ID.
+
+  MODEL_PATH: Path to the model directory. If not specified, the current directory is used by default.
+  """
   from clarifai.runners.server import serve
   serve(model_path)