fix: make linter happy

Signed-off-by: Oleg S <[email protected]>
instructlab · Dec 24, 2024 · 7a8043a · 7a8043a
1 parent d025471
commit 7a8043a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/instructlab/training/config.py b/src/instructlab/training/config.py
@@ -30,8 +30,8 @@ class DeepSpeedOffloadStrategy(Enum):
 
 # public API
 class DistributedBackend(Enum):
-    FSDP: str = "fsdp"
-    DEEPSPEED: str = "deepspeed"
+    FSDP = "fsdp"
+    DEEPSPEED = "deepspeed"
 
 
 # public API
@@ -121,6 +121,7 @@ class DeepSpeedOptions(BaseModel):
     save_samples: int | None = None
 
 
+# public API
 class DistillationConfig(BaseModel):
     """
     Config to use when performing knowledge distillation during training.