From 96fd991fb362c5a253f09244849fdc323b591854 Mon Sep 17 00:00:00 2001 From: Alex Shraer Date: Fri, 20 Dec 2024 07:40:04 +0000 Subject: [PATCH] Add back gpt3 test --- MaxText/tests/gpt3_test.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/MaxText/tests/gpt3_test.py b/MaxText/tests/gpt3_test.py index 4428d7b91..fea40b9e0 100644 --- a/MaxText/tests/gpt3_test.py +++ b/MaxText/tests/gpt3_test.py @@ -55,8 +55,6 @@ def _replace_initialization(key, value): return model_vars -# TODO(b/386317358) -@pytest.mark.skip(reason="Test started failing with pull/1113, skipping for now.") class GPT3(unittest.TestCase): """numerical tests for GPT3."""