idx

.idx/dev.nix

@@ -51,9 +51,6 @@
           python -m venv .venv
           source .venv/bin/activate
           pip install -r requirements.txt
-        '';
-
-        metaflow-local-config = ''
           echo '{}' > ~/.metaflowconfig/config_local.json
         '';
       };

idx-template.json

@@ -3,5 +3,26 @@
     "description": "Building Machine Learning Systems Template",
     "icon": "https://s3.amazonaws.com/www.ml.school/logo.png",
     "params": [
+      {
+        "id": "aws_access_key_id",
+        "name": "AWS Access Key ID",
+        "type": "string",
+        "default": "",
+        "required": false
+      },
+      {
+        "id": "aws_secret_access_key",
+        "name": "AWS Secret Access Key",
+        "type": "string",
+        "default": "",
+        "required": false
+      },
+      {
+        "id": "aws_region",
+        "name": "AWS Region",
+        "type": "string",
+        "default": "us-east-1",
+        "required": false
+      }    
     ]
   }

idx-template.nix

@@ -1,4 +1,4 @@
-{ pkgs, ... }: {
+{ pkgs, aws_access_key_id ? "", aws_secret_access_key ? "", aws_region ? "us-east-1", ... }: {
   channel = "stable-23.11";
 
   bootstrap = ''
@@ -10,8 +10,24 @@
     # Set some permissions
     chmod -R +w "$out"
 
-    # Remove the template files themselves and any connection to the template's
-    # Git repository
-    rm -rf "$out/.git" "$out/idx-template".{nix,json}
+    # Remove the template files
+    rm -rf "$out/idx-template".{nix,json}
+
+    ${
+      if aws_access_key_id != "" then ''
+        mkdir ~/.aws
+
+        cat << EOF >> ~/.aws/credentials
+        [default]
+        aws_access_key_id = ${aws_access_key_id}
+        aws_secret_access_key = ${aws_secret_access_key}    
+        EOF
+
+        cat << EOF >> ~/.aws/config
+        [default]
+        region = ${aws_region}
+        EOF
+      '';
+    }
   '';
 }