DAGWorks-Inc / hamilton

Hamilton helps data scientists and engineers define testable, modular, self-documenting dataflows, that encode lineage/tracing and metadata. Runs and scales everywhere python does.
https://hamilton.dagworks.io/en/latest/
BSD 3-Clause Clear License
1.89k stars 126 forks source link

Update DaskGraphAdapter to mirror RayGraphAdapter #1154

Open skrawcz opened 2 months ago

skrawcz commented 2 months ago

Is your feature request related to a problem? Please describe. We updated the RayGraphAdapter to work with the HamiltonTracker. We should do the same for the DaskGraphAdapter.

Describe the solution you'd like For people using tasks, have it mirror the RayGraphAdapter. For those using dask datatypes -- I don't believe there needs to be any change. For those using both -- we'd need to think through.

Describe alternatives you've considered N/A

Additional context https://github.com/DAGWorks-Inc/hamilton/pull/1103