summaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
-rw-r--r--paramiko/file.py4
-rw-r--r--paramiko/message.py3
-rw-r--r--tests/test_file.py2
3 files changed, 6 insertions, 3 deletions
diff --git a/paramiko/file.py b/paramiko/file.py
index 91579f60..eeb2b8c6 100644
--- a/paramiko/file.py
+++ b/paramiko/file.py
@@ -15,6 +15,8 @@
# You should have received a copy of the GNU Lesser General Public License
# along with Paramiko; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+from io import BytesIO
+
from paramiko.common import (
linefeed_byte_value,
crlf,
@@ -22,7 +24,7 @@ from paramiko.common import (
linefeed_byte,
cr_byte_value,
)
-from paramiko.py3compat import BytesIO, u, bytes_types, text_type
+from paramiko.py3compat import u, bytes_types, text_type
from paramiko.util import ClosingContextManager
diff --git a/paramiko/message.py b/paramiko/message.py
index fd9a2a2f..0927e529 100644
--- a/paramiko/message.py
+++ b/paramiko/message.py
@@ -21,10 +21,11 @@ Implementation of an SSH2 "message".
"""
import struct
+from io import BytesIO
from paramiko import util
from paramiko.common import zero_byte, max_byte, one_byte
-from paramiko.py3compat import long, BytesIO, u
+from paramiko.py3compat import long, u
class Message(object):
diff --git a/tests/test_file.py b/tests/test_file.py
index d4062c02..cbee62ef 100644
--- a/tests/test_file.py
+++ b/tests/test_file.py
@@ -22,10 +22,10 @@ Some unit tests for the BufferedFile abstraction.
import unittest
import sys
+from io import BytesIO
from paramiko.common import linefeed_byte, crlf, cr_byte
from paramiko.file import BufferedFile
-from paramiko.py3compat import BytesIO
from .util import needs_builtin