From 3862790bc83a785fb2a362d1e5e9dcde472946ec Mon Sep 17 00:00:00 2001 From: PGijsbers Date: Thu, 28 Nov 2024 15:07:04 +0200 Subject: [PATCH] Add references property --- amlb/frameworks/definitions.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/amlb/frameworks/definitions.py b/amlb/frameworks/definitions.py index 3484f2104..2f9c545c7 100644 --- 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(