Update appeears.py

earthlab · Jun 5, 2024 · 7dc1093 · 7dc1093
1 parent 20e0b17
commit 7dc1093
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/earthpy/appeears.py b/earthpy/appeears.py
@@ -98,6 +98,8 @@ def __init__(
 		if os.path.exists(self.task_id_path):
 			with open(self.task_id_path, 'r') as task_id_file:
 				self._task_id = task_id_file.readline()
+		elif 'APPEEARS_TASKID' in os.environ:
+			self._task_id = os.environ['APPEEARS_TASKID']
 		else:
 			self._task_id = None
 
@@ -345,4 +347,4 @@ def download_files(self, cache=True):
 						f.write(data)
 
 		# Remove task id file when download is complete
-		os.remove(self.task_id_path)
+		os.remove(self.task_id_path)