Skip to content

Commit

Permalink
DEBUG 0
Browse files Browse the repository at this point in the history
  • Loading branch information
AjayP13 committed Dec 29, 2023
1 parent 926f6fa commit b364e1b
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ jobs:
- name: Setup PROJECT_DATA dir
run: echo -e "\nexport PROJECT_DATA=/tmp/project_data\n" >> ./scripts/project.env
- name: Run tests
run: sudo -H -E env "PATH=$PATH" "LD_LIBRARY_PATH=$LD_LIBRARY_PATH" "HOME=$HOME" ./scripts/run.sh --maxfail=25 --durations=0 --junitxml=./test.xml
run: sudo -H -E env "PATH=$PATH" "LD_LIBRARY_PATH=$LD_LIBRARY_PATH" "HOME=$HOME" ./scripts/run.sh -k "test_in_memory" -s -vvvv --maxfail=25 --durations=0 --junitxml=./test.xml
env:
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
Expand Down
3 changes: 3 additions & 0 deletions src/steps/step_output.py
Original file line number Diff line number Diff line change
Expand Up @@ -755,6 +755,7 @@ def _output_to_dataset( # noqa: C901
) -> OutputDataset | OutputIterableDataset:
# Check whether we are running in the background or not
if pipe: # pragma: no cover
print("IN CHILD")
# We are running in the background
output = __output_to_dataset(
step=step,
Expand Down Expand Up @@ -824,7 +825,9 @@ def dummy_empty_generator():

# Send the data card (sometimes this gets updated in .run())
# + send the pickled version of the OutputDataset to the parent process
print("PIPE PUT BEFORE")
pipe.put(dill.dumps((step._data_card, return_val)))
print("PIPE PUT AFTER")

return return_val # Meaningless, see: pipe.put()
except dill.PicklingError as e:
Expand Down
2 changes: 2 additions & 0 deletions src/utils/background_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -182,7 +182,9 @@ def _run_in_background_process_no_block(
):
p, pipe = run_in_background_process(func, *args, **kwargs)
result_process_func(p)
print("PIPE GET BEFORE")
result_func(pipe.get())
print("PIPE GET AFTER")
p.terminate()

run_in_background_thread(
Expand Down

0 comments on commit b364e1b

Please sign in to comment.