fixed imports

src/rbm/entity.py

@@ -1,6 +1,6 @@
-from params import EntityParams
-from state import RbmState
-from matrix import prob, Mat
+from .params import EntityParams
+from .state import RbmState
+from .matrix import prob, Mat
 
 class Entity:
 	def __init__(self, shape: tuple[int, int], params: EntityParams):

src/rbm/layer.py

@@ -1,9 +1,9 @@
-from params import EntityParams
-from state import RbmState
-from status import Status
-from train import train
-from entity import Entity
-from matrix import Mat, np
+from .params import EntityParams
+from .state import RbmState
+from .status import Status
+from .train import train
+from .entity import Entity
+from .matrix import Mat, np
 
 class Layer:
 	def __init__(self, name: str, shape: tuple[int, int, int, int], params: EntityParams):

src/rbm/stack.py

@@ -1,6 +1,6 @@
 import os
 from enum import Enum
-from layer import Layer
+from .layer import Layer
 
 class StackType(Enum):
 	Deep = "Deep",

src/rbm/stack_deep.py

@@ -1,7 +1,7 @@
-from status import Status
-from train import train
-from stack import Stack, StackType
-from matrix import Mat, np
+from .status import Status
+from .train import train
+from .stack import Stack, StackType
+from .matrix import Mat, np
 
 class StackDeep(Stack):
 	def __init__(self, name: str, work_dir: str = '.'):

src/rbm/stack_factory.py

@@ -1,10 +1,10 @@
 import json
 from collections.abc import Callable
-from stack import StackType
-from layer import Layer
-from params import EntityParams
-from stack_deep import StackDeep
-from stack_rnn import StackRnn
+from .stack import StackType
+from .layer import Layer
+from .params import EntityParams
+from .stack_deep import StackDeep
+from .stack_rnn import StackRnn
 
 class StackFactory:
 	@classmethod

src/rbm/stack_rnn.py

@@ -1,4 +1,4 @@
-from stack import Stack, StackType
+from .stack import Stack, StackType
 
 class StackRnn(Stack):
 	def __init__(self, name: str, work_dir: str = '.'):

src/rbm/state.py

@@ -1,4 +1,4 @@
-from matrix import uniform, Mat, np
+from .matrix import uniform, Mat, np
 
 class RbmState:
 	def __init__(self, w_hv: Mat, b_v: Mat, b_h: Mat):

src/rbm/test_rbm.py

@@ -1,10 +1,10 @@
 import os.path
 import cv2 as cv
 from argparse import ArgumentParser
-from stack_factory import StackFactory
-from status import Status
-from stack_deep import StackDeep
-from matrix import Mat, np, convert
+from .stack_factory import StackFactory
+from .status import Status
+from .stack_deep import StackDeep
+from .matrix import Mat, np, convert
 
 def cv_show(name: str, vec: Mat, shape):
 	img = cv.Mat(convert(np.resize(vec, shape)))

src/rbm/train.py

@@ -1,7 +1,7 @@
 from collections.abc import Callable
-from matrix import gaussian, sample, prob, rms_error_accu, Mat, np
-from entity import Entity
-from status import Status
+from .matrix import gaussian, sample, prob, rms_error_accu, Mat, np
+from .entity import Entity
+from .status import Status
 
 def cd_jens(entity: Entity, v_states: Mat):
 	v_probs = prob(v_states)