Fix imports in snapshot

Signed-off-by: Andre Kurait <akurait@amazon.com>

TrafficCapture/dockerSolution/src/main/docker/migrationConsole/lib/console_link/console_link/models/snapshot.py

@@ -1,11 +1,13 @@
+import argparse
+import datetime
 import logging
 import subprocess
 from abc import ABC, abstractmethod
 from enum import Enum
 from typing import Dict, Optional, Tuple
 
 from cerberus import Validator
-from console_link.models.cluster import Cluster, HttpMethod
+from console_link.models.cluster import AuthMethod, Cluster, HttpMethod
 from console_link.models.command_result import CommandResult
 
 from console_link.models.schema_tools import contains_one_of
@@ -127,6 +129,8 @@ def __init__(self, config: Dict, source_cluster: Cluster, target_cluster: Cluste
 
     def create(self, *args, **kwargs) -> CommandResult:
         assert isinstance(self.target_cluster, Cluster)
+        wait = kwargs.get('wait', False)
+        max_snapshot_rate_mb_per_node = kwargs.get('max_snapshot_rate_mb_per_node', None)
 
         if self.source_cluster.auth_type != AuthMethod.NO_AUTH:
             raise NotImplementedError("Source cluster authentication is not supported for creating snapshots")