test

nationalarchives · Feb 22, 2024 · b082b24 · b082b24
1 parent fbf707f
commit b082b24
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/scripts/check_migrations.py b/.github/scripts/check_migrations.py
@@ -5,10 +5,10 @@
 def list_files_on_branch():
     # Run git diff command to list files on the specified branch
     subprocess.run(['git', 'fetch', 'origin'])
-    current_branch = subprocess.check_output(['git', 'branch', '--show-current']).decode().strip()
-    print(f'Current branch: {current_branch}')
     output = subprocess.check_output(['git', 'diff', '--name-only', 'origin/develop']).decode().strip()
-    return output.split('\n')
+    if not output:
+        output = subprocess.check_output(['git', 'diff', '--name-only', 'origin/main']).decode().strip()
+    return output