configure pytest

PiperOrigin-RevId: 461064163
magenta · Jul 14, 2022 · bdb3b3c · bdb3b3c
1 parent e0f0f21
commit bdb3b3c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -38,7 +38,7 @@ jobs:
       # TODO(adarob): Re-enable once tests are updated.
       run: |
         export FLAX_LAZY_RNG=no
-        pytest
+        pytest mt3/
     # The below step just reports the success or failure of tests as a "commit status".
     # This is needed for copybara integration.
     - name: Report success or failure as github status

diff --git a/mt3/pytest.ini b/mt3/pytest.ini
@@ -0,0 +1,3 @@
+[pytest]
+python_files = *_test.py
+log_level = INFO