graph magic

1 files changed, 9 insertions, 9 deletions

diff --git a/cli/app/utils/cortex_utils.py b/cli/app/utils/cortex_utils.py
index 8f00dfa..bf49f96 100644
--- a/cli/app/utils/cortex_utils.py
+++ b/cli/app/utils/cortex_utils.py
@@ -68,27 +68,27 @@ def fetch_file(url, fn, **kwargs):
         f.write(chunk)
   return size
 
-def upload_fp_to_cortex(opt_folder_id, fp):
+def upload_fp_to_cortex(opt_folder_id, fp, generated='true', module='biggan', activity='invert', datatype='image'):
   """Upload an open file/BytesIO object"""
   files = {
     'file': fp
   }
   data = {
     'folder_id': opt_folder_id,
-    'generated': 'true',
-    'module': 'biggan',
-    'activity': 'invert',
-    'datatype': 'image',
+    'generated': generated,
+    'module': module,
+    'activity': activity,
+    'datatype': datatype,
   }
   url = os.path.join(api_url('folder'), str(opt_folder_id), 'upload/')
   r = requests.post(url, files=files, data=data)
   return None if r.status_code != 200 else r.json()
 
-def upload_bytes_to_cortex(opt_folder_id, fn, fp, mimetype):
+def upload_bytes_to_cortex(opt_folder_id, fn, fp, mimetype, **kwargs):
   """Upload a BytesIO object"""
-  return upload_fp_to_cortex(opt_folder_id, (fn, fp.getvalue(), mimetype,))
+  return upload_fp_to_cortex(opt_folder_id, (fn, fp.getvalue(), mimetype,), **kwargs)
 
-def upload_file_to_cortex(opt_folder_id, fn):
+def upload_file_to_cortex(opt_folder_id, fn, **kwargs):
   """Upload a file from disk"""
   with open(fn, 'rb') as fp:
-    return upload_fp_to_cortex(opt_folder_id, fp)
+    return upload_fp_to_cortex(opt_folder_id, fp, **kwargs)