From 3d7ca5fea2e6012ab85d3c539faf73536f863d40 Mon Sep 17 00:00:00 2001
From: Aparna Jyothi <aparnajyothi-y@github.com>
Date: Fri, 9 May 2025 17:41:53 +0530
Subject: [PATCH] updated code logic for arm64

---
 dist/setup/index.js | 6 ++++++
 src/find-python.ts  | 4 ++++
 2 files changed, 10 insertions(+)

diff --git a/dist/setup/index.js b/dist/setup/index.js
index 98ac87d0..b0c41ae4 100644
--- a/dist/setup/index.js
+++ b/dist/setup/index.js
@@ -96894,12 +96894,18 @@ function useCpythonVersion(version, architecture, updateEnvironment, checkLatest
                     (major > 3 || (major === 3 && minor >= 10))) {
                     versionSuffix += '-32';
                 }
+                else if (architecture === 'arm64') {
+                    versionSuffix += '-arm64';
+                }
                 // Append 't' for freethreaded builds
                 if (freethreaded) {
                     versionSuffix += 't';
                     if (architecture === 'x86-freethreaded') {
                         versionSuffix += '-32';
                     }
+                    else if (architecture === 'arm64-freethreaded') {
+                        versionSuffix += '-arm64';
+                    }
                 }
                 // Add user Scripts path
                 const userScriptsDir = path.join(basePath, 'Python', `Python${versionSuffix}`, 'Scripts');
diff --git a/src/find-python.ts b/src/find-python.ts
index c54edca9..1aef2cec 100644
--- a/src/find-python.ts
+++ b/src/find-python.ts
@@ -164,12 +164,16 @@ export async function useCpythonVersion(
         (major > 3 || (major === 3 && minor >= 10))
       ) {
         versionSuffix += '-32';
+      } else if (architecture === 'arm64') {
+        versionSuffix += '-arm64';
       }
       // Append 't' for freethreaded builds
       if (freethreaded) {
         versionSuffix += 't';
         if (architecture === 'x86-freethreaded') {
           versionSuffix += '-32';
+        } else if (architecture === 'arm64-freethreaded') {
+          versionSuffix += '-arm64';
         }
       }
       // Add user Scripts path