Skip to content

Commit 479c94d

Browse files
style: apply black formatting to fix CI py3.10 check
1 parent 383cabc commit 479c94d

5 files changed

Lines changed: 5 additions & 6 deletions

File tree

src/codefix_env/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,4 +77,4 @@ def __getattr__(name):
7777
from codefix_env.utils.reward_model import RewardMLP
7878

7979
return RewardMLP
80-
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
80+
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

src/codefix_env/rewards.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,4 +155,4 @@ def build_metrics(
155155
default_pipeline = RewardPipeline(
156156
cfg=ScoringConfig(),
157157
neural_model_path=os.environ.get("CODEFIX_REWARD_MODEL_PATH"),
158-
)
158+
)

src/codefix_env/utils/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,4 @@ def __getattr__(name):
3030
from codefix_env.utils.reward_model import RewardMLP
3131

3232
return RewardMLP
33-
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
33+
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

src/codefix_env/utils/metrics.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
import math
1818
from dataclasses import dataclass
1919

20-
2120
# ─────────────────────────────────────────────
2221
# Score Config
2322
# ─────────────────────────────────────────────
@@ -159,4 +158,4 @@ def to_dict(self) -> dict:
159158
"tests_total": self.tests_total,
160159
"total_reward": round(self.total_reward, 4),
161160
"efficiency": round(self.efficiency, 4),
162-
}
161+
}

src/codefix_env/utils/reward_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,4 +76,4 @@ def predict(
7676
dtype=torch.float32,
7777
)
7878
with torch.no_grad():
79-
return self.forward(features).item()
79+
return self.forward(features).item()

0 commit comments

Comments
 (0)