Skip to content

allow custom shape and/or label in to_graphviz()#2987

Merged
mrocklin merged 2 commits intodask:masterfrom
sheppard:custom-dot
Dec 21, 2017
Merged

allow custom shape and/or label in to_graphviz()#2987
mrocklin merged 2 commits intodask:masterfrom
sheppard:custom-dot

Conversation

@sheppard
Copy link
Contributor

  • Tests added / passed
  • Passes flake8 dask
  • Fully documented, including docs/source/changelog.rst for all changes
    and one of the docs/source/*-api.rst files for new API

This extends the data_attributes and function_attributes arguments to support shape and label. Without this change, to_graphviz fails with TypeError: node() got multiple values for keyword argument 'shape'. I don't think these arguments are currently documented so I'm not sure whether new documentation is needed.

@mrocklin
Copy link
Member

I like your implementation here. It looks like it's general to any attribute that we might want to pass through by key. It would be useful to check this generality in tests just to verify that it sticks around in the future.

The failures in tests are unrelated and have since been resolved. They should go away if you end up pushing again.

@mrocklin
Copy link
Member

I found this useful today! Thanks for your work here!

We're waiting on #2973 to merge for tests to pass.

@jcrist
Copy link
Member

jcrist commented Dec 21, 2017

This looks good to me. Anything else left to do here?

@mrocklin mrocklin merged commit f58d456 into dask:master Dec 21, 2017
@mrocklin
Copy link
Member

Thanks for the reminder @jcrist , thanks for the work @sheppard !

@sheppard
Copy link
Contributor Author

Thanks!

@sheppard sheppard deleted the custom-dot branch January 10, 2018 00:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants