fix remove section

ChallengeClub · Oct 7, 2023 · 4704a6b · 4704a6b
1 parent 8e80fdc
commit 4704a6b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/cloud/sam/scripts/post_turtle_training_code/post_turtle_training_code.py b/cloud/sam/scripts/post_turtle_training_code/post_turtle_training_code.py
@@ -49,8 +49,13 @@ def evaluation(event: dict, context):
                 "Content-Type": "text/plain",
             },
         }
-    dir = "/tmp"
-    for f in os.listdir(dir):
-        os.remove(os.path.join(dir, f))
+    directory = "/tmp"
+    for root, dirs, files in os.walk(directory):
+        for file in files:
+            file_path = os.path.join(root, file)
+            os.remove(file_path)
+        for dir in dirs:
+            dir_path = os.path.join(root, dir)
+            os.rmdir(dir_path)
 
     return response
diff --git a/cloud/sam/template.yaml b/cloud/sam/template.yaml
@@ -127,7 +127,6 @@ Resources:
             Method: get
             RestApiId: !Ref TetrisAPI
 
-
   GetResultsFunction:
     Type: AWS::Serverless::Function
     Properties:
@@ -254,7 +253,7 @@ Resources:
       DockerTag: latest
     Properties:
       PackageType: Image
-      Timeout: 5
+      Timeout: 10
       Events:
         PostTetrisTrainingCodeEndpoint:
           Type: Api