fixed imports

Dockerfile

@@ -1,11 +1,24 @@
 FROM python:3.11.4-buster
-RUN  apt-get update -y
-RUN apt-get install -y build-essential
+
+RUN apt-get update -y \
+    && apt-get install -y build-essential \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
 WORKDIR /code
-ADD requirements.txt /code
+
+# Copy requirements and install them first to leverage Docker cache
+COPY requirements.txt /code
 RUN pip install -r requirements.txt
-ADD requirements.yml /code
-ADD ansible.cfg /etc/ansible/
+
+COPY requirements.yml /code
+COPY ansible.cfg /etc/ansible/
 RUN ansible-galaxy install -r requirements.yml
-ADD . /code
+
+# Copy the entire project
+COPY . /code
+
+# Set PYTHONPATH to include the project root
+ENV PYTHONPATH /code
+
 WORKDIR /code/simple_vm_client

simple_vm_client/__init__.py

@@ -1 +0,0 @@
-__all__ = ['ttypes', 'constants', 'VirtualMachineService']

simple_vm_client/forc_connector/template/template.py

@@ -7,8 +7,9 @@
 
 import requests
 import yaml
-from ttypes import ResearchEnvironmentTemplate
-from util.logger import setup_custom_logger
+
+from simple_vm_client.ttypes import ResearchEnvironmentTemplate
+from simple_vm_client.util.logger import setup_custom_logger
 
 # from resenv.backend.Backend import Backend
 
@@ -116,7 +117,6 @@ def update_playbooks(self) -> None:
         for template_metadata in templates_metadata:
             try:
                 if template_metadata.get(NEEDS_FORC_SUPPORT, False):
-
                     metadata = ResearchEnvironmentMetadata(
                         name=template_metadata[TEMPLATE_NAME],
                         port=template_metadata[PORT],