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(