dxpy 0.383.1__tar.gz → 0.385.0__tar.gz

{dxpy-0.383.1 → dxpy-0.385.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dxpy
-Version: 0.383.1
+Version: 0.385.0
 Summary: DNAnexus Platform API bindings for Python
 Home-page: https://github.com/dnanexus/dx-toolkit
 Author: Aleksandra Zalcman, Andrey Kislyuk, Anurag Biyani, Geet Duggal, Katherine Lai, Kurt Jensen, Marek Hrvol, Ohad Rodeh, Phil Sung
@@ -48,6 +48,14 @@ Example:
 $ _DX_DEBUG=1 dx ls
 ```
 
+### Debugging inside the IDE (PyCharm)
+To be able to debug dx-toolkit (dx commands) directly in the IDE, 'Run/Debug Configurations' needs to be changed.
+1. Go to Run → Edit Configurations...
+2. Add New Configuration (Python)
+3. Change script to module (dxpy.scripts.dx)
+4. To Script parameters field write dx command you want to run (eg 'ls' runs 'dx ls')
+5. Apply and OK (now it is possible to start debugging via main() function in dx.py)
+
 Python coding style
 -------------------
 

{dxpy-0.383.1 → dxpy-0.385.0}/Readme.md

@@ -27,6 +27,14 @@ Example:
 $ _DX_DEBUG=1 dx ls
 ```
 
+### Debugging inside the IDE (PyCharm)
+To be able to debug dx-toolkit (dx commands) directly in the IDE, 'Run/Debug Configurations' needs to be changed.
+1. Go to Run → Edit Configurations...
+2. Add New Configuration (Python)
+3. Change script to module (dxpy.scripts.dx)
+4. To Script parameters field write dx command you want to run (eg 'ls' runs 'dx ls')
+5. Apply and OK (now it is possible to start debugging via main() function in dx.py)
+
 Python coding style
 -------------------
 

{dxpy-0.383.1 → dxpy-0.385.0}/dxpy/bindings/dxapplet.py

@@ -58,11 +58,12 @@ class DXExecutable:
             if kwargs.get(arg) is not None:
                 run_input[arg] = kwargs[arg]
 
-        if kwargs.get('instance_type') is not None or kwargs.get('cluster_spec') is not None or kwargs.get('fpga_driver') is not None:
+        if any(kwargs.get(key) is not None for key in ['instance_type', 'cluster_spec', 'fpga_driver', 'nvidia_driver']):
             instance_type_srd = SystemRequirementsDict.from_instance_type(kwargs.get('instance_type'))
             cluster_spec_srd = SystemRequirementsDict(kwargs.get('cluster_spec'))
             fpga_driver_srd = SystemRequirementsDict(kwargs.get('fpga_driver'))
-            run_input["systemRequirements"] = (instance_type_srd + cluster_spec_srd + fpga_driver_srd).as_dict()
+            nvidia_driver_srd = SystemRequirementsDict(kwargs.get('nvidia_driver'))
+            run_input["systemRequirements"] = (instance_type_srd + cluster_spec_srd + fpga_driver_srd + nvidia_driver_srd).as_dict()
 
         if kwargs.get('system_requirements') is not None:
             run_input["systemRequirements"] = kwargs.get('system_requirements')
@@ -195,7 +196,7 @@ class DXExecutable:
             depends_on=None, allow_ssh=None, debug=None, delay_workspace_destruction=None, priority=None, head_job_on_demand=None,
             ignore_reuse=None, ignore_reuse_stages=None, detach=None, cost_limit=None, rank=None, max_tree_spot_wait_time=None,
             max_job_spot_wait_time=None, preserve_job_outputs=None, detailed_job_metrics=None, extra_args=None,
-            fpga_driver=None, system_requirements=None, system_requirements_by_executable=None, **kwargs):
+            fpga_driver=None, system_requirements=None, system_requirements_by_executable=None, nvidia_driver=None, **kwargs):
         '''
         :param executable_input: Hash of the executable's input arguments
         :type executable_input: dict
@@ -252,6 +253,8 @@ class DXExecutable:
         :type system_requirements: dict
         :param system_requirements_by_executable: System requirement by executable double mapping
         :type system_requirements_by_executable: dict
+        :param nvidia_driver: a dict mapping function names to nvidia driver requests
+        :type nvidia_driver: dict
         :rtype: :class:`~dxpy.bindings.dxjob.DXJob`
 
         Creates a new job that executes the function "main" of this executable with
@@ -292,7 +295,8 @@ class DXExecutable:
                                         extra_args=extra_args,
                                         fpga_driver=fpga_driver,
                                         system_requirements=system_requirements,
-                                        system_requirements_by_executable=system_requirements_by_executable)
+                                        system_requirements_by_executable=system_requirements_by_executable,
+                                        nvidia_driver=nvidia_driver)
         return self._run_impl(run_input, **kwargs)
 
 

{dxpy-0.383.1 → dxpy-0.385.0}/dxpy/bindings/dxfile.py

@@ -728,7 +728,7 @@ class DXFile(DXDataObject):
             report_progress_fn(self, len(data))
 
     def wait_until_parts_uploaded(self, **kwargs):
-        self._wait_until_parts_uploaded(self, **kwargs)
+        self._wait_until_parts_uploaded(**kwargs)
 
 
     def get_download_url(self, duration=None, preauthenticated=False, filename=None, project=None, **kwargs):

{dxpy-0.383.1 → dxpy-0.385.0}/dxpy/bindings/dxjob.py

@@ -38,14 +38,15 @@ from ..system_requirements import SystemRequirementsDict
 from ..utils.local_exec_utils import queue_entry_point
 from ..compat import basestring
 
+
 #########
 # DXJob #
 #########
 
-def new_dxjob(fn_input, fn_name, name=None, tags=None, properties=None, details=None,
-              instance_type=None, depends_on=None,
-              cluster_spec=None, fpga_driver=None, system_requirements=None, system_requirements_by_executable=None,
-              **kwargs):
+
+def new_dxjob(fn_input, fn_name, name=None, tags=None, properties=None, details=None, instance_type=None,
+              depends_on=None, cluster_spec=None, fpga_driver=None, system_requirements=None,
+              system_requirements_by_executable=None, nvidia_driver=None, **kwargs):
     '''
     :param fn_input: Function input
     :type fn_input: dict
@@ -71,6 +72,8 @@ def new_dxjob(fn_input, fn_name, name=None, tags=None, properties=None, details=
     :type system_requirements: dict
     :param system_requirements_by_executable: System requirement by executable double mapping
     :type system_requirements_by_executable: dict
+    :param nvidia_driver: a dict mapping function names to nvidia driver requests
+    :type nvidia_driver: dict
     :rtype: :class:`~dxpy.bindings.dxjob.DXJob`
 
     Creates and enqueues a new job that will execute a particular
@@ -94,12 +97,13 @@ def new_dxjob(fn_input, fn_name, name=None, tags=None, properties=None, details=
 
     '''
     dxjob = DXJob()
-    dxjob.new(fn_input, fn_name, name=name, tags=tags, properties=properties,
-              details=details, instance_type=instance_type, depends_on=depends_on,
-              cluster_spec=cluster_spec, fpga_driver=fpga_driver, 
-              system_requirements=system_requirements, system_requirements_by_executable=system_requirements_by_executable, **kwargs)
+    dxjob.new(fn_input, fn_name, name=name, tags=tags, properties=properties, details=details,
+              instance_type=instance_type, depends_on=depends_on, cluster_spec=cluster_spec, fpga_driver=fpga_driver,
+              system_requirements=system_requirements, system_requirements_by_executable=system_requirements_by_executable,
+              nvidia_driver=nvidia_driver, **kwargs)
     return dxjob
 
+
 class DXJob(DXObject):
     '''
     Remote job object handler.
@@ -112,10 +116,9 @@ class DXJob(DXObject):
         DXObject.__init__(self, dxid=dxid)
         self.set_id(dxid)
 
-    def new(self, fn_input, fn_name, name=None, tags=None, properties=None, details=None,
-            instance_type=None, depends_on=None,
-            cluster_spec=None, fpga_driver=None, system_requirements=None, system_requirements_by_executable=None, 
-            **kwargs):
+    def new(self, fn_input, fn_name, name=None, tags=None, properties=None, details=None, instance_type=None,
+            depends_on=None, cluster_spec=None, fpga_driver=None, system_requirements=None,
+            system_requirements_by_executable=None, nvidia_driver=None, **kwargs):
         '''
         :param fn_input: Function input
         :type fn_input: dict
@@ -141,6 +144,8 @@ class DXJob(DXObject):
         :type system_requirements: dict
         :param system_requirements_by_executable: System requirement by executable double mapping
         :type system_requirements_by_executable: dict
+        :param nvidia_driver: a dict mapping function names to nvidia driver requests
+        :type nvidia_driver: dict
 
         Creates and enqueues a new job that will execute a particular
         function (from the same app or applet as the one the current job
@@ -179,11 +184,12 @@ class DXJob(DXObject):
                 req_input["tags"] = tags
             if properties is not None:
                 req_input["properties"] = properties
-            if instance_type is not None or cluster_spec is not None or fpga_driver is not None:
+            if any(requirement is not None for requirement in [instance_type, cluster_spec, fpga_driver, nvidia_driver]):
                 instance_type_srd = SystemRequirementsDict.from_instance_type(instance_type, fn_name)
                 cluster_spec_srd = SystemRequirementsDict(cluster_spec)
                 fpga_driver_srd = SystemRequirementsDict(fpga_driver)
-                req_input["systemRequirements"] = (instance_type_srd + cluster_spec_srd + fpga_driver_srd).as_dict()
+                nvidia_driver_srd = SystemRequirementsDict(nvidia_driver)
+                req_input["systemRequirements"] = (instance_type_srd + cluster_spec_srd + fpga_driver_srd + nvidia_driver_srd).as_dict()
             if system_requirements is not None:
                 req_input["systemRequirements"] = system_requirements
             if system_requirements_by_executable is not None:

{dxpy-0.383.1 → dxpy-0.385.0}/dxpy/scripts/dx.py

@@ -1440,6 +1440,8 @@ def new_project(args):
         inputs["monthlyStorageLimit"] = args.monthly_storage_limit
     if args.default_symlink is not None:
         inputs["defaultSymlink"] = json.loads(args.default_symlink)
+    if args.drive is not None:
+        inputs["drive"] = args.drive
     try:
         resp = dxpy.api.project_new(inputs)
         if args.brief:
@@ -1511,14 +1513,18 @@ def set_visibility(args):
 
 def get_details(args):
     # Attempt to resolve name
-    _project, _folderpath, entity_result = try_call(resolve_existing_path,
+    project, _folderpath, entity_result = try_call(resolve_existing_path,
                                                     args.path, expected='entity')
 
     if entity_result is None:
         err_exit(fill('Could not resolve "' + args.path + '" to a name or ID'), 3)
+    
+    input_params = {}
+    if project is not None:
+        input_params['project'] = project
 
     try:
-        print(json.dumps(dxpy.DXHTTPRequest('/' + entity_result['id'] + '/getDetails', {}), indent=4))
+        print(json.dumps(dxpy.DXHTTPRequest('/' + entity_result['id'] + '/getDetails', input_params), indent=4))
     except:
         err_exit()
 
@@ -3191,10 +3197,12 @@ def run_body(args, executable, dest_proj, dest_path, preset_inputs=None, input_n
         cloned_instance_type = SystemRequirementsDict.from_sys_requirements(cloned_system_requirements, _type='instanceType')
         cloned_cluster_spec = SystemRequirementsDict.from_sys_requirements(cloned_system_requirements, _type='clusterSpec')
         cloned_fpga_driver = SystemRequirementsDict.from_sys_requirements(cloned_system_requirements, _type='fpgaDriver')
+        cloned_nvidia_driver = SystemRequirementsDict.from_sys_requirements(cloned_system_requirements, _type='nvidiaDriver')
         cloned_system_requirements_by_executable = args.cloned_job_desc.get("mergedSystemRequirementsByExecutable", {}) or {}
     else:
         cloned_system_requirements = {}
-        cloned_instance_type, cloned_cluster_spec, cloned_fpga_driver = SystemRequirementsDict({}), SystemRequirementsDict({}), SystemRequirementsDict({})
+        cloned_instance_type, cloned_cluster_spec, cloned_fpga_driver, cloned_nvidia_driver = (
+            SystemRequirementsDict({}), SystemRequirementsDict({}), SystemRequirementsDict({}), SystemRequirementsDict({}))
         cloned_system_requirements_by_executable = {}
 
     # convert runtime --instance-type into mapping {entrypoint:{'instanceType':xxx}}
@@ -3223,12 +3231,15 @@ def run_body(args, executable, dest_proj, dest_path, preset_inputs=None, input_n
     else:
         requested_cluster_spec = cloned_cluster_spec
 
-    # fpga driver now does not have corresponding dx run option, so it can only be requested using the cloned value
+    # fpga/nvidia driver now does not have corresponding dx run option,
+    # so it can only be requested using the cloned value
     requested_fpga_driver = cloned_fpga_driver
+    requested_nvidia_driver = cloned_nvidia_driver
 
-    # combine the requested instance type, full cluster spec, fpga spec
+    # combine the requested instance type, full cluster spec, fpga spec, nvidia spec
     # into the runtime systemRequirements
-    requested_system_requirements = (requested_instance_type + requested_cluster_spec + requested_fpga_driver).as_dict()
+    requested_system_requirements = (requested_instance_type + requested_cluster_spec + requested_fpga_driver +
+                                     requested_nvidia_driver).as_dict()
 
     if (args.instance_type and cloned_system_requirements_by_executable):
         warning = BOLD("WARNING") + ": --instance-type argument: {} may get overridden by".format(args.instance_type) 
@@ -3279,6 +3290,7 @@ def run_body(args, executable, dest_proj, dest_path, preset_inputs=None, input_n
         "instance_type": None,
         "cluster_spec": None,
         "fpga_driver": None,
+        "nvidia_driver": None,
         "stage_instance_types": args.stage_instance_types,
         "stage_folders": args.stage_folders,
         "rerun_stages": args.rerun_stages,
@@ -5804,7 +5816,8 @@ parser_new_project.add_argument('--database-ui-view-only', help='Viewers on the
 parser_new_project.add_argument('--monthly-compute-limit', type=positive_integer, help='Monthly project spending limit for compute')
 parser_new_project.add_argument('--monthly-egress-bytes-limit', type=positive_integer, help='Monthly project spending limit for egress (in Bytes)')
 parser_new_project.add_argument('--monthly-storage-limit', type=positive_number, help='Monthly project spending limit for storage')
-parser_new_project.add_argument('--default-symlink', help='Default symlink for external store account')
+parser_new_project.add_argument('--default-symlink', help='Default symlink for external storage account')
+parser_new_project.add_argument('--drive', help='Drive for external storage account')
 parser_new_project.set_defaults(func=new_project)
 register_parser(parser_new_project, subparsers_action=subparsers_new, categories='fs')
 

{dxpy-0.383.1 → dxpy-0.385.0}/dxpy/scripts/dx_build_app.py

@@ -357,11 +357,11 @@ def _verify_app_source_dir_impl(src_dir, temp_dir, mode, enforce=True):
         if "interpreter" not in manifest['runSpec']:
             raise dxpy.app_builder.AppBuilderException('runSpec.interpreter field was not present')
 
-        if "release" not in manifest['runSpec'] or "distribution" not in manifest['runSpec']:
-            warn_message = 'runSpec.distribution or runSpec.release was not present. These fields '
-            warn_message += 'will be required in a future version of the API. Recommended value '
-            warn_message += 'for distribution is \"Ubuntu\" and release - \"14.04\".'
-            logger.warn(warn_message)
+        if "distribution" not in manifest['runSpec']:
+            raise dxpy.app_builder.AppBuilderException('Required field runSpec.distribution is not present')
+
+        if "release" not in manifest['runSpec']:
+            raise dxpy.app_builder.AppBuilderException('Required field runSpec.release is not present')
 
         if manifest['runSpec']['interpreter'] in ["python2.7", "bash", "python3"]:
             if "file" in manifest['runSpec']:

{dxpy-0.383.1 → dxpy-0.385.0}/dxpy/system_requirements.py

@@ -85,7 +85,7 @@ class SystemRequirementsDict(object):
         It can extract only entrypoints with specific fields ('clusterSpec',
         'instanceType', etc), depending on the value of _type.
         """
-        allowed_types = ['all', 'clusterSpec', 'instanceType', 'fpgaDriver']
+        allowed_types = ['all', 'clusterSpec', 'instanceType', 'fpgaDriver', 'nvidiaDriver']
         if _type not in (allowed_types):
             raise DXError("Expected '_type' to be one of the following: {}".format(allowed_types))
 

dxpy-0.385.0/dxpy/toolkit_version.py

@@ -0,0 +1 @@
+version = '0.385.0'

{dxpy-0.383.1 → dxpy-0.385.0}/dxpy.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dxpy
-Version: 0.383.1
+Version: 0.385.0
 Summary: DNAnexus Platform API bindings for Python
 Home-page: https://github.com/dnanexus/dx-toolkit
 Author: Aleksandra Zalcman, Andrey Kislyuk, Anurag Biyani, Geet Duggal, Katherine Lai, Kurt Jensen, Marek Hrvol, Ohad Rodeh, Phil Sung
@@ -48,6 +48,14 @@ Example:
 $ _DX_DEBUG=1 dx ls
 ```
 
+### Debugging inside the IDE (PyCharm)
+To be able to debug dx-toolkit (dx commands) directly in the IDE, 'Run/Debug Configurations' needs to be changed.
+1. Go to Run → Edit Configurations...
+2. Add New Configuration (Python)
+3. Change script to module (dxpy.scripts.dx)
+4. To Script parameters field write dx command you want to run (eg 'ls' runs 'dx ls')
+5. Apply and OK (now it is possible to start debugging via main() function in dx.py)
+
 Python coding style
 -------------------
 

{dxpy-0.383.1 → dxpy-0.385.0}/test/test_dx_bash_helpers.py

@@ -935,6 +935,22 @@ class TestDXJobutilNewJob(DXTestCase):
                                         "other_function": {"fpgaDriver": "edico-1.4.5"}}}})),
                 {"systemRequirementsByExecutable": {"my_applet":{"main": { "instanceType": "mem2_hdd2_x2", "clusterSpec":{"initialInstanceCount": 3}},
                                         "other_function": { "instanceType": "mem3_ssd2_fpga1_x8", "fpgaDriver": "edico-1.4.5"} }}}),
+            # nvidia driver
+            ("--instance-type-by-executable " +
+             pipes.quote(json.dumps({
+                 "my_applet": {
+                     "main": "mem1_ssd1_v2_x2",
+                     "other_function": "mem2_ssd1_gpu_x16"}})) +
+             " --extra-args " +
+             pipes.quote(json.dumps({
+                 "systemRequirementsByExecutable": {
+                     "my_applet": {
+                         "main": {"instanceType": "mem2_hdd2_x2"},
+                         "other_function": {"nvidiaDriver": "R535"}}}})),
+             {"systemRequirementsByExecutable": {
+                 "my_applet": {"main": {"instanceType": "mem2_hdd2_x2"},
+                               "other_function": {"instanceType": "mem2_ssd1_gpu_x16",
+                                                  "nvidiaDriver": "R535"}}}}),
             # properties - mapping
             (
                 "--property foo=foo_value --property bar=bar_value",

{dxpy-0.383.1 → dxpy-0.385.0}/test/test_dxclient.py

@@ -3332,6 +3332,53 @@ dx-jobutil-add-output record_array $second_record --array
             run("dx run " + applet_id +
                 " --instance-type-by-executable not-a-JSON-string")
 
+    def test_dx_run_clone_nvidia_driver(self):
+        """
+        Run the applet and clone the origin job. Verify nvidiaDriver value.
+        """
+        build_nvidia_version = "R535"
+        run_nvidia_version = "R470"
+
+        applet_id = dxpy.api.applet_new({"project": self.project,
+                                         "dxapi": "1.0.0",
+                                         "runSpec": {"interpreter": "bash",
+                                                     "distribution": "Ubuntu",
+                                                     "release": "20.04",
+                                                     "version": "0",
+                                                     "code": "echo 'hello'",
+                                                     "systemRequirements": {
+                                                         "*": {
+                                                             "instanceType": "mem2_hdd2_x1",
+                                                             "nvidiaDriver": build_nvidia_version
+                                                         }
+                                                     }}
+                                         })['id']
+
+        # Run with unchanged nvidia version (build value)
+        origin_job_id = run(f"dx run {applet_id} --brief -y").strip().split('\n')[-1]
+        origin_job_desc = dxpy.api.job_describe(origin_job_id)
+        assert origin_job_desc["systemRequirements"]["*"]["nvidiaDriver"] == build_nvidia_version
+
+        cloned_job_id = run(f"dx run --clone {origin_job_id} --brief -y").strip()
+        cloned_job_desc = dxpy.api.job_describe(cloned_job_id)
+        assert cloned_job_desc["systemRequirements"]["*"]["nvidiaDriver"] == build_nvidia_version
+
+        # Change nvidia driver version in runtime - origin job (run value)
+        extra_args = json.dumps({"systemRequirements": {"*": {"nvidiaDriver": run_nvidia_version}}})
+        origin_job_id_nvidia_override = run(f"dx run {applet_id} --extra-args '{extra_args}' --brief -y").strip().split('\n')[-1]
+        origin_job_desc = dxpy.api.job_describe(origin_job_id_nvidia_override)
+        assert origin_job_desc["systemRequirements"]["*"]["nvidiaDriver"] == run_nvidia_version
+
+        cloned_job_id_nvidia_override = run(f"dx run --clone {origin_job_id_nvidia_override} --brief -y").strip()
+        cloned_job_desc = dxpy.api.job_describe(cloned_job_id_nvidia_override)
+        assert cloned_job_desc["systemRequirements"]["*"]["nvidiaDriver"] == run_nvidia_version
+
+        # Change nvidia driver version in runtime - cloned job (build value)
+        extra_args = json.dumps({"systemRequirements": {"*": {"nvidiaDriver": build_nvidia_version}}})
+        cloned_job_id_nvidia_override = run(f"dx run --clone {origin_job_id_nvidia_override} --extra-args '{extra_args}' --brief -y").strip()
+        cloned_job_desc = dxpy.api.job_describe(cloned_job_id_nvidia_override)
+        assert cloned_job_desc["systemRequirements"]["*"]["nvidiaDriver"] == build_nvidia_version
+
     def test_dx_run_clone(self):
         applet_id = dxpy.api.applet_new({"project": self.project,
                                          "dxapi": "1.0.0",
@@ -3626,12 +3673,14 @@ dx-jobutil-add-output record_array $second_record --array
         check_new_job_metadata(new_job_desc, orig_job_desc,
                                overridden_fields=['systemRequirements'])
 
-        # fpgaDriver override: new original job with extra_args
+        # fpgaDriver/nvidiaDriver override: new original job with extra_args
         orig_job_id = run("dx run " + other_applet_id +
                           " --instance-count 2 --brief -y " +
                           "--extra-args '" +
-                          json.dumps({"systemRequirements": {"some_ep": {"clusterSpec": {"initialInstanceCount": 12, "bootstrapScript": "z.sh"}, 
-                                                                         "fpgaDriver": "edico-1.4.5"}}}) + "'").strip()
+                          json.dumps({"systemRequirements": {"some_ep":
+                                                {"clusterSpec": {"initialInstanceCount": 12, "bootstrapScript": "z.sh"},
+                                                 "fpgaDriver": "edico-1.4.5",
+                                                 "nvidiaDriver": "R535"}}}) + "'").strip()
         orig_job_desc = dxpy.api.job_describe(orig_job_id)
         check_instance_count(orig_job_desc, ["main", "some_ep","*"], [2, 12, 2])
         # --instance-type and --instance-count override: instance type and cluster spec are resolved independently
@@ -3650,6 +3699,7 @@ dx-jobutil-add-output record_array $second_record --array
         self.assertEqual(new_job_desc['systemRequirements']['*']['instanceType'], 'mem2_hdd2_v2_x2')
         
         self.assertEqual(new_job_desc['systemRequirements']['some_ep']['fpgaDriver'], 'edico-1.4.5')
+        self.assertEqual(new_job_desc['systemRequirements']['some_ep']['nvidiaDriver'], 'R535')
         self.assertEqual(new_job_desc['systemRequirements']['some_ep']['clusterSpec']['bootstrapScript'], 'z.sh')
 
         # --instance-type and --instance-type-by-executable override

{dxpy-0.383.1 → dxpy-0.385.0}/test/test_dxpy.py

@@ -1319,14 +1319,19 @@ def main():
                               "interpreter": "python3",
                               "distribution": "Ubuntu", "release": "20.04",
                               "version": "0",
-                              "execDepends": [{"name": "python-numpy"}]})
+                              "execDepends": [{"name": "python-numpy"}],
+                              "systemRequirements": {
+                                  "*": {
+                                      "nvidiaDriver": "R535"
+                                  }
+                              }})
         dxrecord = dxpy.new_dxrecord()
         dxrecord.close()
         prog_input = {"chromosomes": {"$dnanexus_link": dxrecord.get_id()},
                       "rowFetchChunk": 100}
         dxjob = dxapplet.run(applet_input=prog_input, details={"$dnanexus_link": "hello world"},
                              tags=['foo', '$foo.bar'], properties={'$dnanexus_link.foo': 'barbaz'},
-                             priority="normal")
+                             priority="normal", nvidia_driver={"*": {"nvidiaDriver": "R470"}})
         jobdesc = dxjob.describe()
         self.assertEqual(jobdesc["class"], "job")
         self.assertEqual(jobdesc["function"], "main")
@@ -1346,6 +1351,7 @@ def main():
         self.assertEqual(len(jobdesc["properties"]), 1)
         self.assertEqual(jobdesc["properties"]["$dnanexus_link.foo"], "barbaz")
         self.assertEqual(jobdesc["priority"], "normal")
+        self.assertEqual(jobdesc["systemRequirements"]["*"]["nvidiaDriver"], "R470")
 
         # Test setting tags and properties on job
         dxjob.add_tags(["foo", "bar", "foo"])

dxpy-0.383.1/dxpy/toolkit_version.py

@@ -1 +0,0 @@
-version = '0.383.1'