Merge pull request #186 from python-accelerator-middle-layer/fix-repr-accelerator-and-arrays

Fix repr accelerator and arrays

Author: gubaidulinvadim

pyaml/accelerator.py

@@ -4,7 +4,7 @@
 
 import os
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 
 from .arrays.array import ArrayConfig
 from .common.element import Element
@@ -55,8 +55,8 @@ class ConfigModel(BaseModel):
     simulators: list[Simulator] = None
     data_folder: str
     description: str | None = None
-    arrays: list[ArrayConfig] = None
-    devices: list[Element]
+    arrays: list[ArrayConfig] = Field(default=None, repr=False)
+    devices: list[Element] = Field(repr=False)
 
 
 class Accelerator(object):

pyaml/arrays/array.py

@@ -37,3 +37,8 @@ def __init__(self, cfg: ArrayConfigModel):
 
     def fill_array(self, holder: ElementHolder):
         raise PyAMLException("Array.fill_array() is not subclassed")
+
+    def __repr__(self):
+        # ArrayConfigModel is a super class
+        # ConfigModel is expected from sub classes
+        return repr(self._cfg).replace("ConfigModel", self.__class__.__name__)