minor changes

Author: dickscheid

cli/cli.py

@@ -19,6 +19,8 @@
 import click
 import os
 
+from siibra.retrieval.requests import SiibraHttpRequestError
+
 # ---- Autocompletion
 
 
@@ -143,14 +145,16 @@ def template(ctx, space):
     img.to_filename(fname)
     click.echo(f"Output written to {fname}.")
 
+
 @get.command()
 def ebrainstoken():
     """Retrieve a parcellation map in the given space"""
     import siibra
-    siibra.fetch_ebrains_token()
-    os.environ['HBP_AUTH_TOKEN'] = siibra.EbrainsRequest._KG_API_TOKEN
-    print("Set new EBRAINS KG access token to $HBP_AUTH_TOKEN. The token is:")
-    print(os.environ['HBP_AUTH_TOKEN'])
+    try:
+        siibra.fetch_ebrains_token()
+        print(siibra.EbrainsRequest._KG_API_TOKEN)
+    except SiibraHttpRequestError:
+        exit(1)
 
 
 # ---- Searching for things

siibra/volumes/parcellationmap.py

@@ -740,7 +740,7 @@ class ContinuousParcellationVolume(ParcellationVolume):
 
     # A gitlab instance with holds precomputed sparse indices
     _GITLAB_SERVER = 'https://jugit.fz-juelich.de'
-    _GITLAB_PROJECT = 5779 
+    _GITLAB_PROJECT = 5779
 
     def __init__(self, parcellation, space):
 
@@ -812,7 +812,7 @@ def _load_index(self):
             if conn is None:
                 conn = GitlabConnector(self._GITLAB_SERVER, self._GITLAB_PROJECT, 'main')
                 files = conn.search_files()
-            bname = path.basename(fname) 
+            bname = path.basename(fname)
             if bname in files:
                 logger.debug(f"Retrieving precomputed index for {self.parcellation.name}")
                 raw = conn.get(bname, decode_func=lambda b: b)