fix Bohrium tests (#385)

deepmodeling · Oct 20, 2023 · 16fb0f1 · 16fb0f1
1 parent f844a52
commit 16fb0f1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/test-bohrium.yml b/.github/workflows/test-bohrium.yml
@@ -13,6 +13,8 @@ jobs:
     if: github.repository_owner == 'deepmodeling' && (github.event.label.name == 'Test Bohrium' || github.event_name == 'push')
     steps:
     - uses: actions/checkout@v4
+      with:
+        ref: "${{ github.event.pull_request.merge_commit_sha }}"
     - name: Set up Python 3.12
       uses: actions/setup-python@v4
       with:

diff --git a/tests/test_run_submission_bohrium.py b/tests/test_run_submission_bohrium.py
@@ -22,7 +22,7 @@ def setUp(self):
             remote_profile={
                 "email": os.environ["BOHRIUM_EMAIL"],
                 "password": os.environ["BOHRIUM_PASSWORD"],
-                "project_id": os.environ["BOHRIUM_PROJECT_ID"],
+                "project_id": int(os.environ["BOHRIUM_PROJECT_ID"]),
                 "input_data": {
                     "job_type": "indicate",
                     "log_file": "log",
@@ -54,12 +54,12 @@ def setUp(self):
             accessKey={accesskey}
             """
         ).format(accesskey=os.environ["BOHRIUM_ACCESS_KEY"])
-        Path.home().joinpath(".bohrium").write_text(bohrium_config)
+        Path.home().joinpath(".brmconfig").write_text(bohrium_config)
         self.machine_dict.update(
             batch_type="OpenAPI",
             context_type="OpenAPI",
             remote_profile={
-                "project_id": os.environ["BOHRIUM_PROJECT_ID"],
+                "project_id": int(os.environ["BOHRIUM_PROJECT_ID"]),
                 "machine_type": "c2_m4_cpu",
                 "platform": "ali",
                 "image_address": "registry.dp.tech/dptech/ubuntu:22.04-py3.10",