Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions servicex/func_adl/func_adl_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@
)
from qastle import python_ast_to_text_ast

from func_adl import EventDataset
from func_adl import EventDataset, find_EventDataset
from func_adl.object_stream import S
from servicex.configuration import Configuration
from servicex.dataset import Dataset
Expand Down Expand Up @@ -221,8 +221,9 @@ def set_tree(self, tree_name: str) -> FuncADLDataset[T]:
"""

clone = self.clone_with_new_ast(copy.deepcopy(self.query_ast), self._item_type)
clone._q_ast.args[0].args.append(ast.Str(s="bogus.root"))
clone._q_ast.args[0].args.append(ast.Str(s=tree_name))
c = find_EventDataset(clone.query_ast)
c.args.append(ast.Str(s="bogus.root"))
c.args.append(ast.Str(s=tree_name))
return clone

def generate_qastle(self, a: ast.AST) -> str:
Expand Down