Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion dimos/core/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ def teardown(self, worker) -> None: # type: ignore[no-untyped-def]
import sys

if "cupy" in sys.modules:
import cupy as cp # type: ignore[import-not-found]
import cupy as cp # type: ignore[import-not-found, import-untyped]

# Clear memory pools
mempool = cp.get_default_memory_pool()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@
import glob
import os

from contact_graspnet_pytorch import config_utils # type: ignore[import-not-found]
from contact_graspnet_pytorch.contact_grasp_estimator import ( # type: ignore[import-not-found]
from contact_graspnet_pytorch import config_utils # type: ignore[import-not-found, import-untyped]
from contact_graspnet_pytorch.contact_grasp_estimator import ( # type: ignore[import-not-found, import-untyped]
GraspEstimator,
)
from contact_graspnet_pytorch.data import ( # type: ignore[import-not-found]
from contact_graspnet_pytorch.data import ( # type: ignore[import-not-found, import-untyped]
load_available_input_data,
)
import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion dimos/msgs/sensor_msgs/Image.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@
)

try:
import cupy as cp # type: ignore[import-not-found]
import cupy as cp # type: ignore[import-not-found, import-untyped]
except Exception:
cp = None

Expand Down
4 changes: 2 additions & 2 deletions dimos/msgs/sensor_msgs/image_impls/AbstractImage.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
import rerun as rr

try:
import cupy as cp # type: ignore[import-not-found]
import cupy as cp # type: ignore[import-not-found, import-untyped]

HAS_CUDA = True
except Exception: # pragma: no cover - optional dependency
Expand All @@ -35,7 +35,7 @@
# NVRTC defaults to C++11; libcu++ in recent CUDA requires at least C++17.
if HAS_CUDA:
try:
import cupy.cuda.compiler as _cupy_compiler # type: ignore[import-not-found]
import cupy.cuda.compiler as _cupy_compiler # type: ignore[import-not-found, import-untyped]

if not getattr(_cupy_compiler, "_dimos_force_cxx17", False):
_orig_compile_using_nvrtc = _cupy_compiler.compile_using_nvrtc
Expand Down
4 changes: 2 additions & 2 deletions dimos/msgs/sensor_msgs/image_impls/CudaImage.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@
)

try:
import cupy as cp # type: ignore[import-not-found]
from cupyx.scipy import ( # type: ignore[import-not-found]
import cupy as cp # type: ignore[import-not-found, import-untyped]
from cupyx.scipy import ( # type: ignore[import-not-found, import-untyped]
ndimage as cndimage,
signal as csignal,
)
Expand Down
2 changes: 1 addition & 1 deletion dimos/perception/common/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@

# Optional CuPy support
try: # pragma: no cover - optional dependency
import cupy as cp # type: ignore[import-not-found]
import cupy as cp # type: ignore[import-not-found, import-untyped]

_HAS_CUDA = True
except Exception: # pragma: no cover - optional dependency
Expand Down