@@ -79,4 +79,4 @@ def main():
if __name__ == "__main__":
- main()
+ main()
@@ -312,5 +312,6 @@ def _mp_fn(index):
# For xla_spawn (TPUs)
main()
+