add support for preemptible flag

Author: exdx

paperspace/jobs.py

@@ -77,7 +77,7 @@ def logs(params, tail=False, no_logging=False):
         else:
             try:
                 res = r.json()
-                if 'error' in res:
+                if res in ('error', 'preempted'):
                     if no_logging:
                         return res
                     print_json_pretty(res)
@@ -136,6 +136,7 @@ def waitfor(params):
            or state == 'Error'
            or state == 'Stopped'
            or state == 'Failed'
+           or state == 'Preempted'
            or state == 'Cancelled'):
             return job
         time.sleep(5)

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', 'cluster', 'clusterId', 'ports']:
+                    'workspaceUsername', 'workspacePassword', 'cluster', 'clusterId', 'ports', 'isPreemptible']:
                     if args and not args[0].startswith('--'):
                         params[param] = args.pop(0)
                     else: