Skip to content

Commit e7dff71

Browse files
committed
fix pytype
1 parent 468adc3 commit e7dff71

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

compiler_opt/rl/compilation_runner.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -230,11 +230,12 @@ class CompilationRunnerStub(metaclass=abc.ABCMeta):
230230
"""The interface of a stub to CompilationRunner, for type checkers."""
231231

232232
@abc.abstractmethod
233-
def collect_results(self,
234-
module_spec: corpus.ModuleSpec,
235-
tf_policy_path: str,
236-
collect_default_result: bool,
237-
reward_only: bool = False) -> Tuple[Dict, Dict]:
233+
def collect_results(
234+
self,
235+
module_spec: corpus.ModuleSpec,
236+
tf_policy_path: str,
237+
collect_default_result: bool,
238+
reward_only: bool = False) -> Tuple[Optional[Dict], Optional[Dict]]:
238239
raise NotImplementedError()
239240

240241
@abc.abstractmethod

0 commit comments

Comments
 (0)