Merge pull request #206 from Exabyte-io/feature/SOF-7327

feature/SOF 7327

Author: VsevolodX

examples/job/create_and_submit_job.ipynb

@@ -181,8 +181,9 @@
    },
    "outputs": [],
    "source": [
-    "default_material = material_endpoints.list({\"isDefault\": True, \"owner._id\": ACCOUNT_ID})[0]\n",
-    "default_workflow = workflow_endpoints.list({\"isDefault\": True, \"owner._id\": ACCOUNT_ID})[0]\n",
+    "OWNER_ID = os.getenv(\"ORGANIZATION_ID\") or ACCOUNT_ID\n",
+    "default_material = material_endpoints.list({\"isDefault\": True, \"owner._id\": OWNER_ID})[0]\n",
+    "default_workflow = workflow_endpoints.list({\"isDefault\": True, \"owner._id\": OWNER_ID})[0]\n",
     "\n",
     "material_id = default_material[\"_id\"]\n",
     "workflow_id = default_workflow[\"_id\"]\n",

examples/job/create_and_submit_job.py

@@ -192,7 +192,8 @@
    "source": [
     "# Get some account information\n",
     "project_endpoints = ProjectEndpoints(*ENDPOINT_ARGS)\n",
-    "project_metadata = project_endpoints.list({\"isDefault\": True, \"owner._id\": ACCOUNT_ID})[0]\n",
+    "OWNER_ID = os.getenv(\"ORGANIZATION_ID\") or ACCOUNT_ID\n",
+    "project_metadata = project_endpoints.list({\"isDefault\": True, \"owner._id\": OWNER_ID})[0]\n",
     "project_id = project_metadata[\"_id\"]\n",
     "owner_id = project_metadata[\"owner\"][\"_id\"]\n",
     "\n",

examples/job/get-file-from-job.ipynb

@@ -295,8 +295,9 @@
    },
    "outputs": [],
    "source": [
-    "owner_id = material_endpoints.list({\"isDefault\": True, \"owner._id\": ACCOUNT_ID})[0][\"owner\"][\"_id\"]\n",
-    "project_id = project_endpoints.list({\"isDefault\": True, \"owner._id\": ACCOUNT_ID})[0][\"_id\"]"
+    "OWNER_ID = os.getenv(\"ORGANIZATION_ID\") or ACCOUNT_ID\n",
+    "owner_id = material_endpoints.list({\"isDefault\": True, \"owner._id\": OWNER_ID})[0][\"owner\"][\"_id\"]\n",
+    "project_id = project_endpoints.list({\"isDefault\": True, \"owner._id\": OWNER_ID})[0][\"_id\"]"
    ]
   },
   {
@@ -373,7 +374,7 @@
    "source": [
     "compute = job_endpoints.get_compute(CLUSTER, PPN, NODES, QUEUE, TIME_LIMIT)\n",
     "jobs = job_endpoints.create_by_ids(\n",
-    "    train_materials, band_gap_workflow_id, project_id, owner_id, JOB_NAME_PREFIX, compute\n",
+    "    train_materials, band_gap_workflow_id, project_id, JOB_NAME_PREFIX, owner_id, compute\n",
     ")"
    ]
   },