Skip to content

Type annotation for benchmarks/#7289

Merged
zpcore merged 10 commits intomasterfrom
piz/clean_doc
Jun 20, 2024
Merged

Type annotation for benchmarks/#7289
zpcore merged 10 commits intomasterfrom
piz/clean_doc

Conversation

@zpcore
Copy link
Copy Markdown
Member

@zpcore zpcore commented Jun 15, 2024

Type annotations for benchmarks/.

Also relocated some functions due to circular import.

@zpcore zpcore force-pushed the piz/clean_doc branch 2 times, most recently from a84f2b2 to ba4603d Compare June 16, 2024 07:05
@zpcore zpcore marked this pull request as ready for review June 16, 2024 07:08
@zpcore zpcore requested a review from ysiraichi June 16, 2024 07:44
Copy link
Copy Markdown
Collaborator

@ysiraichi ysiraichi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice PR. I have pointed out a few places where type hints are still missing.
One additional comment: I think it would be better to update the PR title, so that it represents what you are doing here. It doesn't really fixes any document, does it?

Comment thread benchmarks/benchmark_experiment.py Outdated
Comment thread benchmarks/benchmark_experiment.py Outdated
Comment thread benchmarks/benchmark_experiment.py Outdated
Comment thread benchmarks/benchmark_model.py Outdated
Comment thread benchmarks/benchmark_model.py Outdated
Comment thread benchmarks/experiment_runner.py Outdated
Comment thread benchmarks/experiment_runner.py Outdated
Comment thread benchmarks/experiment_runner.py Outdated
Comment thread benchmarks/util.py Outdated
Comment thread benchmarks/experiment_runner.py Outdated
@zpcore zpcore changed the title doc fix Type annotation for benchmarks/ Jun 17, 2024
@zpcore zpcore requested a review from vanbasten23 June 18, 2024 01:07
@zpcore zpcore enabled auto-merge (squash) June 18, 2024 01:07
Comment thread benchmarks/benchmark_experiment.py Outdated
Comment thread benchmarks/benchmark_experiment.py
Comment thread benchmarks/result_analyzer.py
Comment thread benchmarks/torchbench_model.py
Comment thread benchmarks/torchbench_model.py Outdated
@vanbasten23
Copy link
Copy Markdown
Collaborator

Mostly LGTM with minor comments. Thanks!

Comment thread benchmarks/benchmark_experiment.py
Comment thread benchmarks/benchmark_model.py Outdated
Comment thread benchmarks/experiment_runner.py Outdated
Comment thread benchmarks/experiment_runner.py Outdated
@ysiraichi
Copy link
Copy Markdown
Collaborator

Are we also going to start using mypy?

@zpcore
Copy link
Copy Markdown
Member Author

zpcore commented Jun 18, 2024

mypy

@will-cromar and @JackCaoG to see if we have such plan.


def _is_available(self, experiment_config):
def _is_available(self,
experiment_config: List[Dict[str,
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lol, our benchmark script is complicated

@zpcore zpcore merged commit 98dd99e into master Jun 20, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants