modify the environment

linkpark · Sep 14, 2020 · 313f0ed · 313f0ed
1 parent 0b12de2
commit 313f0ed
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/env/mec_offloaing_envs/offloading_task_graph.py b/env/mec_offloaing_envs/offloading_task_graph.py
@@ -117,12 +117,12 @@ class OffloadingTaskGraph(object):
         Args:
             file_name: the raw file of the task graph
         """
-    def __init__(self, file_name, is_matrix=False):
-        self._parse_from_dot(file_name, is_matrix)
+    def __init__(self, file_name):
+        self._parse_from_dot(file_name)
 
     # add task list to
-    def _parse_from_dot(self, file_name, is_matrix):
-        parser = OffloadingDotParser(file_name, is_matrix)
+    def _parse_from_dot(self, file_name):
+        parser = OffloadingDotParser(file_name)
         task_list = parser.generate_task_list()
 
         self.task_number = len(task_list)