- StackFactory removed default constructor

- StackFactory added type hints
This commit is contained in:
2025-12-17 10:53:19 +01:00
parent 437ffbdaad
commit 80ba543323
+4 -7
View File
@@ -1,19 +1,16 @@
import json import json
from collections.abc import Callable from collections.abc import Callable
from stack import StackType from stack import Stack, StackType
from layer import Layer from layer import Layer
from params import RbmParams from params import RbmParams
from stack_deep import StackDeep from stack_deep import StackDeep
from stack_rnn import StackRnn from stack_rnn import StackRnn
class StackFactory: class StackFactory:
def __init__(self):
pass
@classmethod @classmethod
def from_dict(cls, project: dict, layer_constructor: Callable = None): def from_dict(cls, project: dict, layer_constructor: Callable = None) -> Stack:
name = project["stack"]["name"] name = project["stack"]["name"]
layers = project["stack"]["layers"] layers = project["stack"]["layers"]
@@ -59,7 +56,7 @@ class StackFactory:
return obj return obj
@classmethod @classmethod
def from_file(cls, filename: str, layer_constructor: Callable = None): def from_file(cls, filename: str, layer_constructor: Callable = None) -> Stack:
obj = None obj = None
with open(filename, "r") as fp: with open(filename, "r") as fp:
prj = json.load(fp) prj = json.load(fp)