fix register_compute_resource.py

flatironinstitute · Oct 30, 2023 · 18c2d7c · 18c2d7c
1 parent be0c737
commit 18c2d7c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/public/help/dandisets.md b/public/help/dandisets.md
@@ -4,7 +4,7 @@ Search for and select a dandiset.
 
 <!--------------------------------------------------------------------------------->
 {% if staging %}
-You are viewing the staging site.
+You are viewing the staging site. If you need to access embargoed datasets, configure your API key first.
 
 See [Dandi Archive (staging)](https://gui-staging.dandiarchive.org/).
 {% else %}

diff --git a/python/dendro/compute_resource/register_compute_resource.py b/python/dendro/compute_resource/register_compute_resource.py
@@ -74,9 +74,9 @@ def register_compute_resource(*, dir: str, compute_resource_id: Optional[str] =
     print(url)
     print('')
 
-    assert compute_resource_id is not None
-    assert compute_resource_private_key is not None
-    return compute_resource_id, compute_resource_private_key
+    assert COMPUTE_RESOURCE_ID is not None
+    assert COMPUTE_RESOURCE_PRIVATE_KEY is not None
+    return COMPUTE_RESOURCE_ID, COMPUTE_RESOURCE_PRIVATE_KEY
 
 def _random_string(length: int) -> str:
     import random