diff options
-rw-r--r-- | paramiko/file.py | 4 | ||||
-rw-r--r-- | paramiko/message.py | 3 | ||||
-rw-r--r-- | tests/test_file.py | 2 |
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 |