Internal change

PiperOrigin-RevId: 300691387

Author: mn-robot

morph_net/tools/structure_exporter.py

@@ -123,7 +123,7 @@ def save_alive_counts(self, f: IO[bytes]) -> None:
     Args:
       f: a file object where alive counts are saved.
     """
-    f.write(format_structure(self.get_alive_counts()))
+    f.write(format_structure(self.get_alive_counts()))  # pytype: disable=wrong-arg-types
 
   def create_file_and_save_alive_counts(self, base_dir: Text,
                                         global_step: int) -> None: