remove heavy imports (#110)

* add cli

* fix

* fix

* fix

* fix

* test fix

.pre-commit-config.yaml

@@ -36,9 +36,3 @@ repos:
     rev: 0.5.0
     hooks:
       - id: nbstripout
-
-  - repo: https://github.com/myint/docformatter
-    rev: v1.3.1
-    hooks:
-      - id: docformatter
-        args: [ '--in-place' ]

chitra/__init__.py

@@ -1,6 +1,4 @@
 """Deep Learning library for Model Building, Interpretability, Visualization, API Building & Deployment."""
 
-__version__ = "0.1.0"
+__version__ = "0.1.1a0"
 __license__ = "Apache License 2.0"
-
-from chitra.image import Chitra

chitra/cli/__init__.py

@@ -0,0 +1,3 @@
+import typer
+
+app = typer.Typer()

chitra/cli/main.py

@@ -0,0 +1,9 @@
+import typer
+
+from chitra import __version__
+from chitra.cli import app
+
+
+@app.command()
+def hello():
+    typer.echo(f"Hey! You're running chitra version={__version__}")

chitra/image.py

@@ -10,15 +10,15 @@
 
 from chitra.constants import _TF, _TORCH, CHITRA_URL_SEP, IMAGE_CACHE_DIR
 from chitra.coordinates import BoundingBoxes
-from chitra.utility.import_utils import INSTALLED_MODULES
+from chitra.utility.import_utils import is_installed
 
 tf = None
 torch = None
 
-if INSTALLED_MODULES.get(_TF, None):
+if is_installed(_TF):
     import tensorflow as tf
 
-if INSTALLED_MODULES.get(_TORCH, None):
+if is_installed(_TORCH):
     import torch
 
 DATA_FORMATS = Union[str, Image.Image, np.ndarray, "tf.Tensor", "torch.Tensor"]

chitra/imports.py

@@ -0,0 +1,21 @@
+from chitra.constants import _FLAX, _JAX, _TF, _TF_GPU, _TORCH, _TORCHVISION
+from chitra.utility.import_utils import is_installed
+
+INSTALLED_MODULES = {
+    module: is_installed(module)
+    for module in (_TF, _TF_GPU, _TORCH, _TORCHVISION, _JAX, _FLAX)
+}
+
+_FASTAPI_INSTALLED = is_installed("fastapi")
+_UVICORN_INSTALLED = is_installed("uvicorn")
+_PYDANTIC_INSTALLED = is_installed("pydantic")
+_MULTIPART_INSTALLED = is_installed("multipart")
+_SERVE_INSTALLED = (
+    _FASTAPI_INSTALLED
+    and _UVICORN_INSTALLED
+    and _PYDANTIC_INSTALLED
+    and _MULTIPART_INSTALLED
+)
+
+_LOGURU_INSTALLED = is_installed("loguru")
+_RICH_INSTALLED = is_installed("rich")

chitra/logging.py

@@ -1,4 +1,4 @@
-from chitra.utility.import_utils import _LOGURU_INSTALLED
+from chitra.imports import _LOGURU_INSTALLED
 
 if _LOGURU_INSTALLED:
     from loguru import logger

chitra/utility/import_utils.py

@@ -1,27 +1,5 @@
 import importlib
 
-from chitra.constants import _FLAX, _JAX, _TF, _TF_GPU, _TORCH, _TORCHVISION
-
 
 def is_installed(module_name: str):
     return importlib.util.find_spec(module_name) is not None
-
-
-INSTALLED_MODULES = {
-    module: is_installed(module)
-    for module in (_TF, _TF_GPU, _TORCH, _TORCHVISION, _JAX, _FLAX)
-}
-
-_FASTAPI_INSTALLED = is_installed("fastapi")
-_UVICORN_INSTALLED = is_installed("uvicorn")
-_PYDANTIC_INSTALLED = is_installed("pydantic")
-_MULTIPART_INSTALLED = is_installed("multipart")
-_SERVE_INSTALLED = (
-    _FASTAPI_INSTALLED
-    and _UVICORN_INSTALLED
-    and _PYDANTIC_INSTALLED
-    and _MULTIPART_INSTALLED
-)
-
-_LOGURU_INSTALLED = is_installed("loguru")
-_RICH_INSTALLED = is_installed("rich")

pyproject.toml

@@ -56,3 +56,6 @@ test = [
 
 [tool.isort]
 profile = "black"
+
+[tool.flit.scripts]
+chitra = "chitra.cli:app"

tests/serve/test_api.py

@@ -5,7 +5,9 @@
 import pytest
 from fastapi import FastAPI
 
-from chitra.serve import API, create_api, constants as const
+from chitra.serve import API
+from chitra.serve import constants as const
+from chitra.serve import create_api
 
 
 def dummy_model(x):

tests/test_tf_image.py

@@ -2,7 +2,7 @@
 
 import tensorflow as tf
 
-from chitra import Chitra
+from chitra.image import Chitra
 from chitra.tf_image import read_image, resize_image
 
 chitra_banner = (