Skip to content

Commit 7aaded0

Browse files
authored
[hotfix] Ignore mlmodel import if error (#938)
1 parent d58d23e commit 7aaded0

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/viam/module/module.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,12 +56,16 @@
5656
from ..components.switch import Switch # noqa: F401
5757
from ..services.discovery import Discovery # noqa: F401
5858
from ..services.generic import Generic as GenericService # noqa: F401
59-
from ..services.mlmodel import MLModel # noqa: F401
6059
from ..services.motion import Motion # noqa: F401
6160
from ..services.navigation import Navigation # noqa: F401
6261
from ..services.slam import SLAM # noqa: F401
6362
from ..services.vision import Vision # noqa: F401
6463

64+
try:
65+
from ..services.mlmodel import MLModel # noqa: F401
66+
except ImportError:
67+
pass
68+
6569
from .service import ModuleRPCService
6670
from .types import Reconfigurable, Stoppable
6771

@@ -73,7 +77,7 @@ def _parse_module_args() -> argparse.Namespace:
7377
p = argparse.ArgumentParser(description="Start this viam python module")
7478
p.add_argument("socket_path", help="path where this module will serve a unix socket")
7579
p.add_argument("--log-level", type=lambda name: pylogging._nameToLevel[name.upper()], default=logging.INFO)
76-
p.add_argument("--tcp-mode", action='store_true')
80+
p.add_argument("--tcp-mode", action="store_true")
7781
return p.parse_known_args()[0]
7882

7983

0 commit comments

Comments
 (0)