Update src/maxtext path

PiperOrigin-RevId: 860276641

Author: SurbhiJainUSC

codecov.yml

@@ -39,8 +39,8 @@ ignore:
   - "src/MaxText/inference"
   - "src/MaxText/inference_mlperf"
   - "src/MaxText/scratch_code"
-  - "src/MaxText/distillation" # code moved to src/MaxText/trainers/post_train/distillation
-  - "src/MaxText/sft" # code moved to src/MaxText/trainers/post_train/sft
+  - "src/MaxText/distillation" # code moved to src/maxtext/trainers/post_train/distillation
+  - "src/MaxText/sft" # code moved to src/maxtext/trainers/post_train/sft
 
 
 flags:

pyproject.toml

@@ -38,7 +38,7 @@ Repository = "https://github.com/AI-Hypercomputer/maxtext.git"
 allow-direct-references = true
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/MaxText", "src/MaxText", "src/install_maxtext_extra_deps"]
+packages = ["src/MaxText", "src/maxtext", "src/install_maxtext_extra_deps"]
 
 [tool.hatch.build.targets.wheel.hooks.custom]
 path = "build_hooks.py"

src/MaxText/distillation/train_distill.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Shim for Distillation Trainer in `src/MaxText/trainers/post_train/distillation`."""
+"""Shim for Distillation Trainer in `src/maxtext/trainers/post_train/distillation`."""
 
 import sys
 import importlib

src/MaxText/sft/sft_trainer.py

@@ -12,7 +12,7 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-"""Shim for SFT Trainer in `src/MaxText/trainers/post_train/sft`."""
+"""Shim for SFT Trainer in `src/maxtext/trainers/post_train/sft`."""
 
 import sys
 import importlib