fixed type issues

luxonis · Sep 10, 2024 · 6ead271 · 6ead271
1 parent da8430e
commit 6ead271
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -82,7 +82,7 @@ jobs:
         cache: pip
 
     - name: Install dependencies
-      run: pip install .
+      run: pip install .[dev]
 
     - uses: jakebailey/pyright-action@v2
       with:

diff --git a/luxonis_train/__main__.py b/luxonis_train/__main__.py
@@ -180,7 +180,7 @@ def archive(
 
 def version_callback(value: bool):
     if value:
-        typer.echo(f"LuxonisTrain Version: {version(__package__)}")
+        typer.echo(f"LuxonisTrain Version: {version('luxonis_train')}")
         raise typer.Exit()
 
 

diff --git a/luxonis_train/callbacks/metadata_logger.py b/luxonis_train/callbacks/metadata_logger.py
@@ -57,6 +57,8 @@ def _get_editable_package_git_hash(
         """
         try:
             distribution = pkg_resources.get_distribution(package_name)
+            if distribution.location is None:
+                return None
             package_location = osp.join(distribution.location, package_name)
 
             # remove any additional folders in path (e.g. "/src")

diff --git a/luxonis_train/nodes/backbones/efficientnet.py b/luxonis_train/nodes/backbones/efficientnet.py
@@ -1,7 +1,7 @@
 from typing import Any
 
 import torch
-from torch import Tensor
+from torch import Tensor, nn
 
 from luxonis_train.nodes.base_node import BaseNode
 
@@ -36,7 +36,7 @@ def __init__(
         """
         super().__init__(**kwargs)
 
-        self.backbone = torch.hub.load(  # type: ignore
+        self.backbone: nn.Module = torch.hub.load(  # type: ignore
             "rwightman/gen-efficientnet-pytorch",
             "efficientnet_lite0",
             pretrained=download_weights,