mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
9115df2895
commit
2be7442dd5
9 changed files with 26 additions and 48 deletions
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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']}")
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue