diff --git a/src/tilegym/ops/__init__.py b/src/tilegym/ops/__init__.py index 83df0a4a..ec2e7516 100644 --- a/src/tilegym/ops/__init__.py +++ b/src/tilegym/ops/__init__.py @@ -4,6 +4,7 @@ """TileGym ops module - contains all operation interfaces and backend implementations""" +import warnings from tilegym.backend import is_backend_available # Backend implementations @@ -16,7 +17,6 @@ try: from . import cutile except (ImportError, RuntimeError): - import warnings warnings.warn("Cutile backend import failed, cutile operations will not be available") cutile = None # type: ignore @@ -29,7 +29,6 @@ try: from . import triton except (ImportError, RuntimeError): - import warnings warnings.warn("Triton backend import failed, triton operations will not be available") triton = None # type: ignore