Minor fix.

Tune.ipynb

@@ -41,9 +41,6 @@
     "  '--model', 'AstroCnnModel',\n",
     "  '--config_name', 'local_global',\n",
     "  '--train_files', '',\n",
-    "  '--model_dir', '',\n",
-    "  '--cloud_project_id', 'mdan-playground',\n",
-    "  '--cloud_user', 'mdan',\n",
     "])\n",
     "\n",
     "client = tune.initialize_client()"
@@ -155,6 +152,8 @@
    "outputs": [],
    "source": [
     "import difflib\n",
+    "import pprint\n",
+    "from astronet import models\n",
     "\n",
     "config1 = models.get_model_config('AstroCNNModel', 'local_global_multiclass_extra_features')\n",
     "for param in params[219]:\n",
@@ -215,7 +214,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.5"
+   "version": "3.7.6"
   }
  },
  "nbformat": 4,

astronet/tune.py

@@ -28,6 +28,7 @@
 import json
 import logging
 import time
+improt os
 import pprint
 import sys
 
@@ -92,18 +93,6 @@
     help="OAuth secrets file, see https://github.com/googleapis/"
     "google-api-python-client/blob/master/docs/client-secrets.md.")
 
-parser.add_argument(
-    "--cloud_user",
-    type=str,
-    required=True,
-    help="Cloud user name.")
-
-parser.add_argument(
-    "--cloud_project_id",
-    type=str,
-    required=True,
-    help="Cloud project ID.")
-
 parser.add_argument(
     "--study_id",
     type=str,
@@ -120,9 +109,11 @@
 
 REGION = 'us-central1'
 
+CLOUD_PROJECT_ID = os.environ["CLOUD_PROJECT_ID"]
+
 
 def study_parent():
-  return 'projects/{}/locations/{}'.format(FLAGS.cloud_project_id, REGION)
+  return 'projects/{}/locations/{}'.format(CLOUD_PROJECT_ID, REGION)
 
 
 def study_id():
@@ -143,7 +134,7 @@ def trial_name(trial_id):
 
 def operation_name(operation_id):
   return 'projects/{}/locations/{}/operations/{}'.format(
-      FLAGS.cloud_project_id, REGION, operation_id)
+      CLOUD_PROJECT_ID, REGION, operation_id)
 
 
 def study_config(config):
@@ -169,7 +160,7 @@ def initialize_client():
   appflow.run_console()
   credentials = appflow.credentials
 
-  client = storage.Client(FLAGS.cloud_project_id)
+  client = storage.Client(CLOUD_PROJECT_ID)
   bucket = client.get_bucket('caip-optimizer-alpha-public')
   blob = bucket.get_blob('api/alpha-ml_google_rest_v1.json')
   service = blob.download_as_string()