Skip to content

Commit

Permalink
just pring a log msg for now
Browse files Browse the repository at this point in the history
  • Loading branch information
douglasdavis committed Nov 10, 2023
1 parent 67655fe commit 6e46fdf
Showing 1 changed file with 14 additions and 17 deletions.
31 changes: 14 additions & 17 deletions src/dask_awkward/lib/io/io.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from __future__ import annotations

import functools
import logging
import math
from collections.abc import Callable, Iterable
from dataclasses import dataclass
Expand Down Expand Up @@ -49,6 +50,9 @@
from dask_awkward.lib.core import Array


logger = logging.getLogger(__name__)


class _FromAwkwardFn:
def __init__(self, arr: ak.Array) -> None:
self.arr = arr
Expand Down Expand Up @@ -478,28 +482,21 @@ def wrapped(*args, **kwargs):
try:
return fn(*args, **kwargs)
except allowed_exceptions as err:
if Queue is not None and get_worker is not None:
try:
_ = get_worker()
queue = Queue("dak_returned_empty")
queue.put((args, kwargs, str(err)))
except ValueError:
pass

logmsg = (
"%s call failed with args %s and kwargs %s; empty array returned. %s"
% (
str(fn),
str(args),
str(kwargs),
str(err),
)
)
logger.info(logmsg)
return fn.mock_empty(backend)

return wrapped


def returned_empty_report() -> list[Any]:
report = []
if Queue is not None:
queue = Queue("dak_returned_empty")
while queue.qsize():
report.append(queue.get())
return report


def from_map(
func: Callable,
*iterables: Iterable,
Expand Down

0 comments on commit 6e46fdf

Please sign in to comment.