Skip to content

Commit 8c4d0b0

Browse files
SurbhiJainUSCGoogle-ML-Automation
SurbhiJainUSC
authored and
Google-ML-Automation
committed
Move managed_mldiagnostics to src/maxtext/common
PiperOrigin-RevId: 862929999
1 parent 25f09a2 commit 8c4d0b0

12 files changed

Lines changed: 2 additions & 2 deletions

src/MaxText/managed_mldiagnostics.py renamed to src/maxtext/common/managed_mldiagnostics.py

File renamed without changes.

‎src/maxtext/common/metric_logger.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@
2727

2828
import google_cloud_mldiagnostics as mldiag
2929

30-
from MaxText.managed_mldiagnostics import ManagedMLDiagnostics
3130
from MaxText.globals import EPS
3231
from maxtext.common.gcp_workload_monitor import GCPWorkloadMonitor
32+
from maxtext.common.managed_mldiagnostics import ManagedMLDiagnostics
3333
from maxtext.utils import gcs_utils
3434
from maxtext.utils import max_logging
3535
from maxtext.utils import max_utils

‎src/maxtext/common/profiler.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424
import google_cloud_mldiagnostics as mldiag
2525

26-
from MaxText.managed_mldiagnostics import ManagedMLDiagnostics
26+
from maxtext.common.managed_mldiagnostics import ManagedMLDiagnostics
2727
from maxtext.utils import max_logging
2828

2929

src/MaxText/utils/exceptions.py renamed to src/maxtext/utils/exceptions.py

File renamed without changes.

src/MaxText/utils/gcs_utils.py renamed to src/maxtext/utils/gcs_utils.py

File renamed without changes.

src/MaxText/utils/lora_utils.py renamed to src/maxtext/utils/lora_utils.py

File renamed without changes.

src/MaxText/utils/max_logging.py renamed to src/maxtext/utils/max_logging.py

File renamed without changes.

src/MaxText/utils/max_utils.py renamed to src/maxtext/utils/max_utils.py

File renamed without changes.

src/MaxText/utils/maxtext_utils.py renamed to src/maxtext/utils/maxtext_utils.py

File renamed without changes.

src/MaxText/utils/model_creation_utils.py renamed to src/maxtext/utils/model_creation_utils.py

File renamed without changes.

0 commit comments

Comments
 (0)