Skip to content

Commit 25f09a2

Browse files
SurbhiJainUSCGoogle-ML-Automation
authored andcommitted
Move decode.py to src/maxtext
PiperOrigin-RevId: 862915762
1 parent 80f3efc commit 25f09a2

3 files changed

Lines changed: 2 additions & 2 deletions

File tree

tests/integration/decode_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from absl.testing import absltest
2424
from contextlib import redirect_stdout
2525

26-
from MaxText.decode import main as decode_main
26+
from maxtext.decode import main as decode_main
2727
from MaxText.globals import MAXTEXT_PKG_DIR, MAXTEXT_ASSETS_ROOT
2828

2929
pytestmark = pytest.mark.integration_test

tests/integration/generate_param_only_checkpoint_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from MaxText.globals import MAXTEXT_ASSETS_ROOT, MAXTEXT_PKG_DIR
2424
from MaxText.train import main as train_main
2525
from MaxText.generate_param_only_checkpoint import main as generate_param_only_ckpt_main
26-
from MaxText.decode import main as decode_main
26+
from maxtext.decode import main as decode_main
2727
from tests.integration.checkpointing_test import get_checkpointing_command
2828

2929

0 commit comments

Comments
 (0)