rcsb-embedding-model 0.0.30__tar.gz → 0.0.31__tar.gz

{rcsb_embedding_model-0.0.30 → rcsb_embedding_model-0.0.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rcsb-embedding-model
-Version: 0.0.30
+Version: 0.0.31
 Summary: Protein Embedding Model for Structure Search
 Project-URL: Homepage, https://github.com/rcsb/rcsb-embedding-model
 Project-URL: Issues, https://github.com/rcsb/rcsb-embedding-model/issues
@@ -11,6 +11,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
 Requires-Dist: esm>=3.2.0
+Requires-Dist: importlib-metadata>=8.7.0
 Requires-Dist: lightning>=2.5.0
 Requires-Dist: typer>=0.15.0
 Description-Content-Type: text/markdown

{rcsb_embedding_model-0.0.30 → rcsb_embedding_model-0.0.31}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rcsb-embedding-model"
-version = "0.0.30"
+version = "0.0.31"
 authors = [
     { name="Joan Segura", email="joan.segura@rcsb.org" },
 ]
@@ -14,6 +14,7 @@ classifiers = [
 license = "BSD-3-Clause"
 license-files = ["LICEN[CS]E*"]
 dependencies=[
+    "importlib-metadata >= 8.7.0",
     "esm >= 3.2.0",
     "lightning >= 2.5.0",
     "typer >= 0.15.0"

rcsb_embedding_model-0.0.31/src/rcsb_embedding_model/__init__.py

@@ -0,0 +1,9 @@
+from importlib_metadata import version, PackageNotFoundError
+from rcsb_embedding_model.rcsb_structure_embedding import RcsbStructureEmbedding
+
+try:
+    __version__ = version("rcsb-embedding-model")
+except PackageNotFoundError:
+    __version__ = "0.0.0"
+
+__all__ = ["RcsbStructureEmbedding", "__version__"]

{rcsb_embedding_model-0.0.30 → rcsb_embedding_model-0.0.31}/src/rcsb_embedding_model/cli/inference.py

@@ -3,6 +3,7 @@ from typing import Annotated, List
 
 import typer
 
+from rcsb_embedding_model import __version__
 from rcsb_embedding_model.cli.args_utils import arg_devices
 from rcsb_embedding_model.types.api_types import StructureFormat, Accelerator, SrcLocation, SrcProteinFrom, \
     StructureLocation, SrcAssemblyFrom, SrcTensorFrom, OutFormat
@@ -437,5 +438,22 @@ def complete_embedding(
     )
 
 
+def version_callback(value: bool):
+    if value:
+        typer.echo(f"{__version__}")
+        raise typer.Exit()
+
+@app.callback()
+def main(
+        version: bool = typer.Option(
+            None,
+            "--version",
+            callback=version_callback,
+            is_eager=True,
+            help="Show the version and exit",
+        )
+):
+    pass
+
 if __name__ == "__main__":
     app()

rcsb_embedding_model-0.0.30/src/rcsb_embedding_model/__init__.py

@@ -1,4 +0,0 @@
-
-from rcsb_embedding_model.rcsb_structure_embedding import RcsbStructureEmbedding
-
-__all__ = ["RcsbStructureEmbedding"]