fix status check loop

KevKibe · Oct 29, 2024 · abf8f0d · abf8f0d
1 parent c5233c9
commit abf8f0d
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 18 deletions.
diff --git a/.github/workflows/test-kaggle-action.yml b/.github/workflows/test-kaggle-action.yml
@@ -22,4 +22,5 @@ jobs:
           title: "My Test Kernel"
           custom_script: "print('Success')"
           enable_internet: true
-          enable_gpu: false
+          enable_gpu: true
+          enable_tpu: true
diff --git a/action.yml b/action.yml
@@ -183,24 +183,25 @@ runs:
         
         metadata=$(jq -r '. | tojson' < "$json_path")
         
-        title="${{ inputs.title }}"  
-        formatted_title=$(echo "$title" | tr '[:upper:]' '[:lower:]' | sed 's/ /_/g')
+        kernel_name="${{ inputs.title }}"  
+        kernel_slug=$(echo "kernel_name" | tr '[:upper:]' '[:lower:]' | sed 's/ /_/g')
         
         log_message "Updating metadata with specified inputs..."
         metadata=$(echo "$metadata" | jq --arg title "${{ inputs.title }}" \
-                                 --arg code_file "$OUTPUT_NOTEBOOK" \
-                                 --argjson enable_gpu "${{ inputs.enable_gpu }}" \
-                                 --argjson enable_internet "${{ inputs.enable_internet }}" \
-                                 --argjson enable_tpu "${{ inputs.enable_tpu }}" \
-                                 '. |
-                                   .id = (.id | split("/")[0]) + "/" + $title |
-                                   .title = $title |
-                                   .code_file = $code_file |
-                                   .language = "python" |
-                                   .kernel_type = "notebook" |
-                                   .enable_gpu = $enable_gpu |
-                                   .enable_internet = $enable_internet |
-                                   .enable_tpu = $enable_tpu')
+                                         --arg id "kernel_slug" \
+                                         --arg code_file "$OUTPUT_NOTEBOOK" \
+                                         --argjson enable_gpu "${{ inputs.enable_gpu }}" \
+                                         --argjson enable_internet "${{ inputs.enable_internet }}" \
+                                         --argjson enable_tpu "${{ inputs.enable_tpu }}" \
+                                         '. |
+                                           .id = $id |
+                                           .title = $title |
+                                           .code_file = $code_file |
+                                           .language = "python" |
+                                           .kernel_type = "notebook" |
+                                           .enable_gpu = $enable_gpu |
+                                           .enable_internet = $enable_internet |
+                                           .enable_tpu = $enable_tpu')
         
         echo "Saving updated metadata back to $json_path"
         echo "$metadata" | jq '.' > "$json_path"
@@ -219,7 +220,9 @@ runs:
       run: |
         
         username="${{ inputs.username }}"
-        kernel_name="${username}/${{ inputs.title }}"
+        raw_kernel_name="${{ inputs.title }}"
+        kernel_slug=$(echo "$raw_kernel_name" | tr '[:upper:]' '[:lower:]' | sed 's/ /_/g')
+        kernel_name="${username}/${kernel_slug}"
         action_path="${{ github.action_path }}"
         action_path=$(realpath "${{ github.action_path }}")
         echo "Checking status for Kaggle kernel: '$kernel_name'"
@@ -255,4 +258,8 @@ runs:
             kaggle kernels output "$kernel_name" 2>&1
             log_message "::endgroup::"
             exit 1
-        fi
+        fi
+
+  branding:
+    icon: activity
+    color: green