-
Notifications
You must be signed in to change notification settings - Fork 3.6k
[WIP] Fix mypy typing for utilities.auto_restart
#9454
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
utilities.auto_restartutilities.auto_restart
|
This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 7 days if no further activity occurs. If you need further help see our docs: https://pytorch-lightning.readthedocs.io/en/latest/generated/CONTRIBUTING.html#pull-request or ask the assistance of a core contributor here or on Slack. Thank you for your contributions. |
| if not isinstance(dataloader.dataset, CaptureIterableDataset): | ||
| fast_forward_sampler = _find_fast_forward_samplers(dataloader) | ||
| if fast_forward_sampler is not None: | ||
| out.update(fast_forward_sampler.state_dict(num_batches_processed=num_batches_processed)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Won't update works ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It returns me the following mypy error on my local:
pytorch_lightning/utilities/auto_restart.py:421: error: Argument 1 to "update" of "dict" has incompatible type "Dict[int, Dict[str, int]]"; expected "Mapping[Union[int, str], Union[Dict[str, int], int, None]]" [arg-type]
|
This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 7 days if no further activity occurs. If you need further help see our docs: https://pytorch-lightning.readthedocs.io/en/latest/generated/CONTRIBUTING.html#pull-request or ask the assistance of a core contributor here or on Slack. Thank you for your contributions. |
|
This pull request is going to be closed. Please feel free to reopen it create a new from the actual master. |
Still, WIP and some mypy issues need to be fixed.
What does this PR do?
Part of #7037. This PR disentangles large #7218. It fixes mypy typing in
utilities.auto_restart.pyDoes your PR introduce any breaking changes? If yes, please list them.
None
Before submitting
PR review
Anyone in the community is welcome to review the PR.
Before you start reviewing make sure you have read Review guidelines. In short, see the following bullet-list:
Did you have fun?
Make sure you had fun coding 🙃