Skip to content

Commit

Permalink
CI
Browse files Browse the repository at this point in the history
  • Loading branch information
jakep-allenai committed Feb 14, 2025
1 parent c05e015 commit 25ec87b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/main.yml
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ jobs:
- name: Setup Python
uses: actions/setup-python@v4
with:
python-version: "3.10"
python-version: "3.11"

- name: Install requirements
run: |
Expand Down
1 change: 1 addition & 0 deletions tests/test_dataprep.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,6 +183,7 @@ def testListTargetAnchorLength(self):
self.assertEqual(zero_count + full_count, num_iterations, "Total count should equal number of iterations")


@pytest.mark.nonci
class TestMolmoDataPrep(unittest.TestCase):
def testMolmoDefaultSetup(self):
processor = AutoProcessor.from_pretrained("allenai/Molmo-7B-O-0924", trust_remote_code=True, torch_dtype="auto", device_map="auto")
Expand Down

0 comments on commit 25ec87b

Please sign in to comment.