Skip to content

Commit 0149112

Browse files
committed
fix fields
1 parent 0c9d1d8 commit 0149112

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

autointent/nodes/schemes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def generate_models_and_union_type_for_classes(
2222
globalns = getattr(cls.from_context, "__globals__", {})
2323
type_hints = get_type_hints(cls.from_context, globalns, None) # Resolve forward refs
2424

25-
fields = {"module_name": (Literal[cls.name], Field(default=cls.name))}
25+
fields = {"module_name": (Literal[cls.name], Field(...))}
2626

2727
for param_name, param in init_signature.parameters.items():
2828
if param_name in ("self", "cls", "context"):

tests/assets/configs/multilabel.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
- node_type: embedding
2-
metric: scoring_accuracy
2+
metric: retrieval_hit_rate
33
search_space:
44
- module_name: logreg
55
cv: [2]

tests/pipeline/test_inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
@pytest.mark.parametrize(
99
"task_type",
10-
["multiclass", "multilabel", "description"],
10+
["multilabel"],
1111
)
1212
def test_inference_config(dataset, task_type):
1313
project_dir = setup_environment()

0 commit comments

Comments
 (0)