Pass ruff check

PrimeIntellect-ai · Jan 10, 2025 · be3ef20 · be3ef20
1 parent 818e671
commit be3ef20
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/src/zeroband/config.py b/src/zeroband/config.py
@@ -1,5 +1,3 @@
-
-
 from typing import Literal
 
 from pydantic import model_validator

diff --git a/src/zeroband/data.py b/src/zeroband/data.py
@@ -417,7 +417,7 @@ def load_all_datasets(
         split_world_size = world_size
 
 
-    logger.info(f"Loading Train dataset(s)")
+    logger.info("Loading Train dataset(s)")
 
     ds = _load_datasets(
         dataset_names=data_config.dataset_name_or_paths,

diff --git a/src/zeroband/train.py b/src/zeroband/train.py
@@ -17,7 +17,7 @@
 from zeroband.comms import ElasticDeviceMesh
 from zeroband.loss import cross_entropy_max_z_loss
 from zeroband.models.llama.model import create_block_mask_from_seqlens
-from zeroband.config import Config, MemoryProfilerConfig
+from zeroband.config import Config  #, MemoryProfilerConfig
 
 from zeroband.utils import (
     FakeTokenizer,
Original file line number	Diff line number	Diff line change
		@@ -1,5 +1,3 @@


		from typing import Literal

		from pydantic import model_validator
Expand Down