Turn Docker into an optional dependency

gqlalchemy/instance_runner.py

@@ -20,13 +20,17 @@
 from enum import Enum
 from typing import Any, Dict, Union
 
-import docker
+try:
+    import docker
+except ModuleNotFoundError:
+    docker = None
 import psutil
 
 from gqlalchemy.exceptions import (
     GQLAlchemyWaitForConnectionError,
     GQLAlchemyWaitForDockerError,
     GQLAlchemyWaitForPortError,
+    raise_if_not_imported,
 )
 from gqlalchemy.vendors.memgraph import Memgraph
 
@@ -86,7 +90,7 @@ def wait_for_port(
 
 
 def wait_for_docker_container(
-    container: "docker.Container", delay: float = 0.01, timeout: float = 5.0, backoff: int = 2
+    container, delay: float = 0.01, timeout: float = 5.0, backoff: int = 2
 ) -> None:
     """Wait for a Docker container to enter the status `running`.
 
@@ -239,6 +243,8 @@ class MemgraphInstanceDocker(MemgraphInstance):
     def __init__(
         self, docker_image: DockerImage = DockerImage.MEMGRAPH, docker_image_tag: str = DOCKER_IMAGE_TAG_LATEST, **data
     ) -> None:
+        raise_if_not_imported(dependency=docker, dependency_name="docker")
+
         super().__init__(**data)
         self.docker_image = docker_image
         self.docker_image_tag = docker_image_tag