[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-01-26 16:41:22 +00:00
parent 9115df2895
commit 2be7442dd5
9 changed files with 26 additions and 48 deletions

View file

@ -84,10 +84,8 @@ class TaskDataModule(LightningDataModule):
if not data:
raise NotImplementedError("Each task has its own data set")
if not self.batch_size or self.batch_size < 1:
raise NotImplementedError(
"Each task configures its own batch_size. \
Please set it explicitly, to avoid confusion."
)
raise NotImplementedError("Each task configures its own batch_size. \
Please set it explicitly, to avoid confusion.")
def train_dataloader(self):
self.check_dataset(self.data_train)

View file

@ -92,8 +92,7 @@ def send_code(code: str):
def main():
"""Entrypoint for local execution. Accepts code as input."""
# Example usage: pass code as a string argument
return_obj = send_code.remote(
"""
return_obj = send_code.remote("""
import torch
import math
@ -144,8 +143,7 @@ for step in range(20):
print(f"Step {step + 1}: x = {x.item():.4f}, loss = {loss.item():.4f}")
print(f"\\nOptimal x: {x.item():.4f}")
"""
)
""")
print("\n===== Optimizer Execution Result =====")
print(f"Filename: {return_obj['filename']}")

View file

@ -34,8 +34,7 @@ class OptimizerEvaluator:
categories=DiscreteScale(["yes", "no"]),
explanation=False,
)
.prompt(
"""
.prompt("""
You are an expert code validator specializing in PyTorch optimizers.
Your task is to determine if the provided optimizer code is completely valid and error-free.
@ -57,8 +56,7 @@ class OptimizerEvaluator:
- "no" if ANY criterion fails or there are ANY potential issues
Be extremely strict in your evaluation.
"""
)
""")
.via("xai/grok-3-latest", retries=2)
)