Update robot.py

dshokouhi · Dec 5, 2018 · 3b6c36e · 3b6c36e
1 parent 82a1400
commit 3b6c36e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pybotvac/robot.py b/pybotvac/robot.py
@@ -8,7 +8,7 @@
 # Disable warning due to SubjectAltNameWarning in certificate
 requests.packages.urllib3.disable_warnings()
 
-SUPPORTED_SERVICES = ['basic-1', 'minimal-2', 'basic-2', 'basic-3']
+SUPPORTED_SERVICES = ['basic-1', 'minimal-2', 'basic-2', 'basic-3', 'basic-4']
 
 
 class UnsupportedDevice(Exception):
@@ -78,7 +78,7 @@ def start_cleaning(self, mode=2, navigation_mode=1, category=None):
                         'mode': mode,
                         'modifier': 1}
                     }
-        elif self.service_version == 'basic-3':
+        elif self.service_version == 'basic-3' or 'basic-4':
             json = {'reqId': "1",
                     'cmd': "startCleaning",
                     'params': {