mdk - npm Package Compare versions

Comparing version 5.7.3 to 5.8.0

mdk.egg-info/entry_points.txt

 [console_scripts]
 mdk = mdk.main:mdk
-

mdk.egg-info/PKG-INFO

 Metadata-Version: 2.1
 Name: mdk
-Version: 5.7.3
+Version: 5.8.0
 Summary: a docker-compose helper
@@ -9,3 +9,2 @@ Home-page: https://matician.com/
 License: MIT
-Platform: UNKNOWN
 Classifier: Environment :: Console
@@ -48,3 +47,1 @@ Classifier: Intended Audience :: Developers
 * `~/.config/mdk/mdk.json`
-
-

mdk/main.py

@@ -16,2 +16,10 @@ import click
 
+    def nvidia_option(f):
+        def callback(ctx, param, value):
+            backend.nvidia = value
+        return click.option(
+            '--nvidia',
+            default=False, is_flag=True, expose_value=False,
+            callback=callback)(f)
+
     @click.group()
@@ -24,2 +32,3 @@ @click.version_option(version=VERSION)
     @nogpu_option
+    @nvidia_option
     def mdk_bash():
@@ -35,2 +44,3 @@ backend.start(implicit=True)
     @nogpu_option
+    @nvidia_option
     @click.option("--interactive/--non-interactive", default=True, is_flag=True)
@@ -96,2 +106,3 @@ @click.option("--tty/--no-tty", default=True, is_flag=True)
     @nogpu_option
+    @nvidia_option
     def mdk_sh():
@@ -119,2 +130,3 @@ backend.start(implicit=True)
     @nogpu_option
+    @nvidia_option
     def mdk_up():
@@ -125,2 +137,3 @@ backend.start(ensure_up_to_date=True)
     @nogpu_option
+    @nvidia_option
     def mdk_zsh():
@@ -127,0 +140,0 @@ backend.start(implicit=True)

mdk/utils.py

@@ -74,2 +74,3 @@ """mdk utility functions & classes"""
     no_gpu = False
+    nvidia = False
 
@@ -278,2 +279,7 @@ def __init__(self):
 
+        if self.nvidia:
+            opt_builder.extend(
+                ["-e", "NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics"]
+            )
+
         # old images required --tty to start and run properly
@@ -327,3 +333,4 @@ if self.conf_version < (5, 0):
     def cmd(self, *args: str, quiet=False) -> None:
-        cmd = ['docker'] + [(self.container_name() if a == '@CONTAINER@' else a) for a in args]
+        docker = 'nvidia-docker' if self.nvidia else 'docker'
+        cmd = [docker] + [(self.container_name() if a == '@CONTAINER@' else a) for a in args]
         Log.cmd(cmd)
@@ -330,0 +337,0 @@ code = subprocess.run(cmd, stdout=subprocess.PIPE if quiet else None).returncode

PKG-INFO

 Metadata-Version: 2.1
 Name: mdk
-Version: 5.7.3
+Version: 5.8.0
 Summary: a docker-compose helper
@@ -9,3 +9,2 @@ Home-page: https://matician.com/
 License: MIT
-Platform: UNKNOWN
 Classifier: Environment :: Console
@@ -48,3 +47,1 @@ Classifier: Intended Audience :: Developers
 * `~/.config/mdk/mdk.json`
-
-

Alert delta unavailable

Currently unable to show alert delta for PyPI packages.

Improved metrics

Total package byte prevSize

27325

1.99%

Lines of code

507

3.68%