Add references property

openml · Nov 28, 2024 · 3862790 · 3862790
1 parent e8ab966
commit 3862790
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/amlb/frameworks/definitions.py b/amlb/frameworks/definitions.py
@@ -4,7 +4,7 @@
 import itertools
 import logging
 import os
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import List, Optional, Union, TYPE_CHECKING
 
 from amlb.utils import Namespace, config_load, str_sanitize
@@ -256,15 +256,17 @@ class Framework:
     abstract: bool
     module: str
     version: str
-    params: dict
     # Image
     image: Image
     # Setup
-    setup_env: dict
-    setup_args: list[str]
     _setup_cmd: str | None
     setup_cmd: str | None
     setup_script: str | None
+    setup_env: dict = field(default_factory=dict)
+    setup_args: list[str] = field(default_factory=list)
+    # core but optional
+    params: dict = field(default_factory=dict)
+    refs: list[str] = field(default_factory=list)
 
 
 def load_framework_definition(