write torchscript to temp file instead bytesio in model test

Author: kazhang

test/test_models.py

@@ -1,13 +1,13 @@
 import contextlib
 import functools
-import io
 import operator
 import os
 import pkgutil
 import sys
 import traceback
 import warnings
 from collections import OrderedDict
+from tempfile import TemporaryDirectory
 
 import pytest
 import torch
@@ -126,10 +126,10 @@ def assert_export_import_module(m, args):
 
         def get_export_import_copy(m):
             """Save and load a TorchScript model"""
-            buffer = io.BytesIO()
-            torch.jit.save(m, buffer)
-            buffer.seek(0)
-            imported = torch.jit.load(buffer)
+            with TemporaryDirectory() as dir:
+                path = os.path.join(dir, "script.pt")
+                m.save(path)
+                imported = torch.jit.load(path)
             return imported
 
         m_import = get_export_import_copy(m)