add cluster+clusterId to param dictionary

Author: exdx

paperspace/main.py

@@ -75,7 +75,7 @@ def main():
                 param = opt[2:]
                 if param in ['script', 'python', 'conda', 'ignoreFiles', 'apiKey', 'container', 'machineType', 'name',
                     'project', 'projectId', 'command', 'workspace', 'dataset', 'registryUsername', 'registryPassword',
-                    'workspaceUsername', 'workspacePassword']:
+                    'workspaceUsername', 'workspacePassword', 'cluster', 'clusterId']:
                     if args and not args[0].startswith('--'):
                         params[param] = args.pop(0)
                     else:

tests/test.py

@@ -23,7 +23,7 @@ def errorcheck(res):
 for job in jobs:
     print(job['id'])
 
-print("jobs.create({'project': '%s', 'machineType': 'P5000', 'container': 'paperspace/tensorflow-python', 'command': './do.sh', 'workspace': '~/myproject3'})" % project)
+print("jobs.create({'project': '%s', 'machineType': 'P5000', 'container': 'paperspace/tensorflow-python', 'command': './do.sh', 'workspace': '~/myproject3', 'cluster': 'Gradient-Node'})" % project)
 job = paperspace.jobs.create({'project': project,
                               'machineType': 'P5000', 'container': 'paperspace/tensorflow-python',
                               'command': './do.sh', 'workspace': '~/myproject3'})