Try this?

src/template.py

@@ -11,14 +11,21 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+import os
 
 TEMPLATES = ["initial", "initial_with_metrics", "accelerate"]
 
+def _get_filename(template: str) -> str:
+    """
+    Takes an template and returns the respective filename relative to the cwd.
+    """
+    return os.path.join(os.getcwd(), "code_samples", template)
+
 def get_templates() -> dict:
     """
     Returns a dictionary of template type to code content
     """
     return {
-        template: open(f"../code_samples/{template}").read()
+        template: open(_get_filename(template)).read()
         for template in TEMPLATES
     }