Skip to content
Snippets Groups Projects
Commit b2ed4f21 authored by Maxim Gonchar's avatar Maxim Gonchar
Browse files

feat: `storage.read_paths()` and `storage.plot()`

parent a2d48779
No related branches found
No related tags found
No related merge requests found
from multikeydict.nestedmkdict import NestedMKDict from multikeydict.nestedmkdict import NestedMKDict
from multikeydict.visitor import NestedMKDictVisitor from multikeydict.visitor import NestedMKDictVisitor
from dagflow.output import Output
from typing import Union, Tuple, List, Optional, Dict from typing import Union, Tuple, List, Optional
from tabulate import tabulate from tabulate import tabulate
from pandas import DataFrame from pandas import DataFrame
...@@ -18,6 +19,23 @@ def trunc(text: str, width: int) -> str: ...@@ -18,6 +19,23 @@ def trunc(text: str, width: int) -> str:
return '\n'.join(line[:width] for line in text.split('\n')) return '\n'.join(line[:width] for line in text.split('\n'))
class ParametersStorage(NestedMKDict): class ParametersStorage(NestedMKDict):
def read_paths(self) -> None:
for key, value in self.walkitems():
labels = getattr(value, 'labels', None)
if labels is None:
continue
key = '.'.join(key)
labels.setdefault('paths', []).append(key)
def plot(self) -> None:
from dagflow.plot import plot_auto
for key, value in self.walkitems():
if not isinstance(value, Output):
continue
plot_auto(value)
def to_list(self, **kwargs) -> list: def to_list(self, **kwargs) -> list:
return self.visit(ParametersVisitor(kwargs)).data_list return self.visit(ParametersVisitor(kwargs)).data_list
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment