joris
/
maximum-weight-matching
1
0
Fork 0
Code Pull Requests Activity

Add missing type annotation

This commit is contained in:
Joris van Rantwijk 2023-03-11 17:32:49 +01:00
parent 5048bbaf99
commit ba28bbf7ca
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/generate/make_slow_graph.py
View File

@ -10,7 +10,7 @@ from __future__ import annotations
import sys
import argparse
from typing import TextIO
from typing import Any, TextIO
count_make_blossom = [0]
@ -26,18 +26,17 @@ def patch_matching_code() -> None:
orig_substage_calc_dual_delta = (
mwmatching._MatchingContext.substage_calc_dual_delta)
def stub_make_blossom(*args, **kwargs):
def stub_make_blossom(*args: Any, **kwargs: Any) -> Any:
count_make_blossom[0] += 1
return orig_make_blossom(*args, **kwargs)
def stub_substage_calc_dual_delta(*args, **kwargs):
def stub_substage_calc_dual_delta(*args: Any, **kwargs: Any) -> Any:
count_delta_step[0] += 1
ret = orig_substage_calc_dual_delta(*args, **kwargs)
# print("DELTA", ret)
return ret
mwmatching._MatchingContext.make_blossom = stub_make_blossom
mwmatching._MatchingContext.substage_calc_dual_delta = (
mwmatching._MatchingContext.make_blossom = stub_make_blossom # type: ignore
mwmatching._MatchingContext.substage_calc_dual_delta = ( # type: ignore
stub_substage_calc_dual_delta)