Enable --overriding-parameter-count-checks for ptype and fix related failures (#216)

Author: mtrofin

.github/workflows/main.yml

@@ -38,7 +38,7 @@ jobs:
         task: [Typecheck, Lint, Yapf, Test]
         include:
           - task: Typecheck
-            cmd: pytype -j auto .
+            cmd: pytype -j auto --overriding-parameter-count-checks .
           - task: Lint
             cmd: pylint --rcfile .pylintrc --recursive yes .
           - task: Yapf

compiler_opt/rl/local_data_collector_test.py

@@ -89,7 +89,11 @@ def mock_collect_data(loaded_module_spec: corpus.LoadedModuleSpec, policy,
 class Sleeper(compilation_runner.CompilationRunner):
   """Test CompilationRunner that just sleeps."""
 
-  def collect_data(self, loaded_module_spec, policy, reward_stat, model_id):
+  def collect_data(self,
+                   loaded_module_spec,
+                   policy=None,
+                   reward_stat=None,
+                   model_id=None):
     _ = loaded_module_spec, policy, reward_stat
     compilation_runner.start_cancellable_process(['sleep', '3600s'], 3600,
                                                  self._cancellation_manager)

compiler_opt/tools/generate_default_trace_test.py

@@ -36,7 +36,11 @@
 class MockCompilationRunner(compilation_runner.CompilationRunner):
   """A compilation runner just for test."""
 
-  def collect_data(self, loaded_module_spec, policy, reward_stat, model_id):
+  def collect_data(self,
+                   loaded_module_spec,
+                   policy=None,
+                   reward_stat=None,
+                   model_id=None):
     sequence_example_text = """
       feature_lists {
         feature_list {