cleanup

Author: prashantmital

pymongo/network.py

@@ -40,7 +40,7 @@
 _UNPACK_HEADER = struct.Struct("<iiii").unpack
 
 
-def command(sock_info, max_wire_version, dbname, spec, slave_ok, is_mongos,
+def command(sock_info, dbname, spec, slave_ok, is_mongos,
             read_preference, codec_options, session, client, check=True,
             allowable_errors=None, address=None,
             check_keys=False, listeners=None, max_bson_size=None,
@@ -157,7 +157,8 @@ def command(sock_info, max_wire_version, dbname, spec, slave_ok, is_mongos,
                 client._process_response(response_doc, session)
             if check:
                 helpers._check_command_response(
-                    response_doc, max_wire_version, None, allowable_errors,
+                    response_doc, sock_info.max_wire_version, None,
+                    allowable_errors,
                     parse_write_concern_error=parse_write_concern_error)
     except Exception as exc:
         if publish:

pymongo/pool.py

@@ -680,7 +680,7 @@ def command(self, dbname, spec, slave_ok=False,
         if self.op_msg_enabled:
             self._raise_if_not_writable(unacknowledged)
         try:
-            return command(self, self.max_wire_version, dbname, spec,
+            return command(self, dbname, spec,
                            slave_ok, self.is_mongos, read_preference,
                            codec_options, session, client, check,
                            allowable_errors, self.address, check_keys,