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

clarifai/__init__.py

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

clarifai/cli/__main__.py~

@@ -0,0 +1,4 @@
+from my_cli_tool.cli import main
+
+if __name__ == "__main__":
+  main()

clarifai/cli/model.py

@@ -66,6 +66,31 @@ def download_checkpoints(model_path, out_path, stage):
   builder.download_checkpoints(stage=stage, checkpoint_path_override=out_path)
 
 
+@model.command()
+@click.option(
+    '--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',