[WIP] Sync community fork changes

.github/workflows/ci.yml

@@ -0,0 +1,34 @@
+name: CI
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v3
+
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.10'
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -e ".[dev]"
+        pip install types-cffi
+
+    - name: Lint with ruff
+      run: ruff check .
+
+    - name: Type check with mypy
+      run: mypy axengine
+
+    - name: Test with pytest
+      run: pytest -m "not hardware" --cov=axengine --cov-report=term-missing

.gitignore

@@ -0,0 +1,3 @@
+.sisyphus/
+__pycache__/
+.coverage

.pre-commit-config.yaml

@@ -0,0 +1,13 @@
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.1.9
+    hooks:
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.8.0
+    hooks:
+      - id: mypy
+        args: [--python-version=3.8]

axengine/__init__.py

@@ -8,15 +8,38 @@
 # thanks to community contributors list below:
 #   zylo117: https://github.com/zylo117, first implementation of the axclrt backend
 
-from ._providers import axengine_provider_name, axclrt_provider_name
-from ._providers import get_all_providers, get_available_providers
+import os
+
+from ._logging import get_logger
+from ._node import NodeArg as NodeArg
+from ._providers import (
+    axclrt_provider_name as axclrt_provider_name,
+)
+from ._providers import (
+    axengine_provider_name as axengine_provider_name,
+)
+from ._providers import (
+    get_all_providers as get_all_providers,
+)
+from ._providers import (
+    get_available_providers,
+)
+from ._session import InferenceSession as InferenceSession
+from ._session import SessionOptions as SessionOptions
+
+logger = get_logger(__name__)
 
-# check if axclrt is installed, or is a supported chip(e.g. AX650, AX620E etc.)
 _available_providers = get_available_providers()
-if not _available_providers:
-    raise ImportError(
-        f"No providers found. Please make sure you have installed one of the following: {get_all_providers()}")
-print("[INFO] Available providers: ", _available_providers)
+_is_test_or_ci = bool(os.getenv("CI") or os.getenv("PYTEST_CURRENT_TEST"))
 
-from ._node import NodeArg
-from ._session import SessionOptions, InferenceSession
+if not _available_providers:
+    _provider_error_message = (
+        "No execution providers available. Install the required hardware libraries "
+        "(ax_engine or axcl_rt) and check that the target hardware/driver is available."
+    )
+    if _is_test_or_ci:
+        logger.warning(_provider_error_message)
+    else:
+        raise ImportError(_provider_error_message)
+else:
+    logger.info("Available providers: %s", _available_providers)