Skip to content

Commit 367a71a

Browse files
committed
catch op container up with API
1 parent 5881b53 commit 367a71a

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

data_algebra/op_container.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -78,15 +78,22 @@ def transform(
7878
data_model=None
7979
):
8080
assert isinstance(self.ops, data_algebra.data_ops.ViewRepresentation)
81-
return self.ops.transform(
82-
X=X,
83-
data_model=data_model
84-
)
81+
assert not isinstance(X, OpC)
82+
if isinstance(X, data_algebra.data_ops_types.OperatorPlatform):
83+
self.set(self.ops.transform(X=X, data_model=data_model))
84+
return self
85+
# assume a table
86+
return self.ops.transform(X=X, data_model=data_model)
8587

8688
# noinspection PyPep8Naming
8789
def act_on(self, X, *, data_model=None):
88-
self.set(self.ops.act_on(X=X, data_model=data_model))
89-
return self
90+
assert not isinstance(X, OpC)
91+
if isinstance(X, data_algebra.data_ops_types.OperatorPlatform):
92+
self.set(self.ops.act_on(X=X, data_model=data_model))
93+
return self
94+
# assume a table
95+
self.ops.act_on(X=X, data_model=data_model)
96+
return
9097

9198
def replace_leaves(self, replacement_map: Dict[str, Any]):
9299
self.set(self.ops.replace_leaves(replacement_map))

0 commit comments

Comments
 (0)