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