Add remote env var

ecmwf-lab · Feb 26, 2024 · 0869080 · 0869080
1 parent 061a70c
commit 0869080
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ai_models/__main__.py b/ai_models/__main__.py
@@ -225,6 +225,7 @@ def _main(argv):
         help="Enable remote execution, read url and token from ~/.config/ai-models/api.yaml",
         action="store_true",
         dest="remote_execution",
+        default=(os.environ.get("AI_MODELS_REMOTE", "0") == "1"),
     )
 
     args, unknownargs = parser.parse_known_args(argv)