Skip to content

Commit 05dbb0d

Browse files
committed
Merge branch 'master' into origin_master
# Conflicts: # coconut/astar.coco # coconut/auto_data.coco # coconut/convert.coco # coconut/omni_converter.coco # data_tree/__init__.py # data_tree/_series.py # data_tree/coconut/astar.py # data_tree/coconut/auto_data.py # data_tree/coconut/convert.py # data_tree/coconut/omni_converter.py # data_tree/ops/cache.py # data_tree/storage_manager.py # data_tree/util.py # test/coconut/__init__.py # test/test_conversion.py
2 parents c64b394 + 0723039 commit 05dbb0d

File tree

134 files changed

+16876
-4227
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

134 files changed

+16876
-4227
lines changed

coconut/calc.py

Lines changed: 0 additions & 890 deletions
This file was deleted.

coconut/controllable.coco

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ class ZippedCW(Sourced):
183183

184184
class VizCW(Sourced):
185185
def __init__(self,src,visualizer:object->"displayable"):
186-
self.src = src
186+
self.value = src
187187
self.visualizer = visualizer
188188
@lazy
189189
def value(self):

coconut/monad.coco

Lines changed: 0 additions & 14 deletions
This file was deleted.

coconut/parsetab.py

Lines changed: 0 additions & 37 deletions
This file was deleted.
File renamed without changes.

coconut/visualization.coco

Lines changed: 0 additions & 132 deletions
This file was deleted.

coconut_test/test_conversion.coco

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ start = str_to_img_def("numpy,float32,HWC,RGB,0_1")
1111
end = str_to_img_def("torch,uint8,BHWC,RGB,0_255")
1212
data END from ImageDef
1313
data DUMMY from ImageDef
14+
from archpainter.rulebook import legacy_auto as auto
1415
def dummy_rule(imdef):
1516
return [(a->a,end,1,"dummy")]
1617

@@ -95,7 +96,7 @@ def test_casting():
9596

9697
def test_omni_converter():
9798
from data_tree.coconut.omni_converter import auto_img,cast_imdef_str_to_imdef,cast_imdef_to_imdef_str
98-
from data_tree.coconut.auto_data import AutoData
99+
from omni_converter.coconut.auto_data import AutoData
99100
x = np.ones((100,100,3),dtype="float32")
100101
auto_x:AutoData = auto_img("numpy,float32,HW,L,0_1")(x)
101102
assert (auto_x.to("numpy,float32,HW,L,0_255") == 255).all()
@@ -111,12 +112,10 @@ def test_omni_converter():
111112
#assert format == n_format,f"{format} != {n_format}"
112113

113114
def test_tuple_conversion():
114-
from data_tree import auto
115115
from archpainter.experiments.rgba2xyz_mod.instances import torch_xyz, pix2pix_rgb_batch, TORCH_XYZ_BATCH
116116
logger.info(auto(("image,RGB,RGB","image,RGB,RGB"))((None,None)).converter(("pix2pix_batch,nc=1", TORCH_XYZ_BATCH)))
117117

118118
def test_rgb_to_yuv():
119-
from data_tree import auto
120119
from archpainter.experiments.rgba2xyz_mod.instances import torch_xyz, pix2pix_rgb_batch, TORCH_XYZ_BATCH
121120
logger.info(auto("image,RGB,RGB")(None).converter("image,YCbCr,YCbCr"))
122121

coconut_test/test_mp_queue.coco

Lines changed: 0 additions & 74 deletions
This file was deleted.

data_tree/__init__.py

Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
<<<<<<< HEAD
12

23

34

@@ -29,3 +30,76 @@ def auto_img(codec):
2930
def _l(img):
3031
return AutoImage(img, codec,solver)
3132
return _l
33+
=======
34+
from omni_converter.auto_data.auto_v2 import RuledData
35+
from omni_converter.solver.rules import AutoRuleBook
36+
37+
38+
def managed_cache(root_dir):
39+
from data_tree.cache import ConditionedFilePathProvider
40+
return ConditionedFilePathProvider(root_dir)
41+
42+
43+
def series(iterable):
44+
from data_tree._series import Series
45+
return Series.from_iterable(iterable)
46+
47+
48+
def unlist_auto(items):
49+
from data_tree.coconut.omni_converter import unlist
50+
return unlist(items)
51+
52+
53+
# def auto(format) -> Callable[[Any], RuledData]:
54+
# return auto_img(format)
55+
56+
57+
def resolve_format(value, format_resolution_rule: AutoRuleBook):
58+
from omni_converter.auto_data.auto_v2 import AutoData2
59+
return AutoData2(value=value, format=value).with_rules(format_resolution_rule).to("format")
60+
61+
62+
def ruled(value,
63+
rules: AutoRuleBook,
64+
format=None,
65+
format_resolution_rule: AutoRuleBook = None) -> "RuledData":
66+
if format_resolution_rule is None:
67+
from omni_converter.auto_data.format_resolution import FORMAT_RESOLUTION_RULES
68+
format_resolution_rule = FORMAT_RESOLUTION_RULES
69+
if format is None:
70+
# try to automatically get format from value given rules
71+
format = resolve_format(value, format_resolution_rule)
72+
from omni_converter.auto_data.auto_v2 import AutoData2
73+
return AutoData2(value=value, format=format).with_rules(rules)
74+
75+
76+
class PicklableLogger:
77+
def info(self, message):
78+
self.logger.info(message)
79+
80+
def warning(self, message):
81+
self.logger.warning(message)
82+
83+
def error(self, message):
84+
self.logger.error(message)
85+
86+
def debug(self, message):
87+
self.logger.debug(message)
88+
89+
def verbose(self, message):
90+
self.logger.verbose(message)
91+
92+
def __getstate__(self):
93+
return None
94+
95+
def __setstate__(self, state):
96+
from loguru import logger
97+
self.logger = logger
98+
99+
def __init__(self):
100+
from loguru import logger
101+
self.logger = logger
102+
103+
104+
logger = PicklableLogger()
105+
>>>>>>> master

0 commit comments

Comments
 (0)