Skip to content

Commit

Permalink
fix pyright
Browse files Browse the repository at this point in the history
  • Loading branch information
Han Wang committed Feb 2, 2024
1 parent cc8a7a2 commit f281d1d
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 15 deletions.
15 changes: 0 additions & 15 deletions dpgen2/exploration/task/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,18 +54,3 @@ def files(self) -> Dict:
The dict storing all files for the task. The file name is a key of the dict, and the file content is the corresponding value.
"""
return self._files


class FooTask(ExplorationTask):
def __init__(
self,
conf_name="conf.lmp",
conf_cont="",
inpu_name="in.lammps",
inpu_cont="",
):
super().__init__()
self._files = {
conf_name: conf_cont,
inpu_name: inpu_cont,
}
15 changes: 15 additions & 0 deletions dpgen2/exploration/task/task_group.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,21 @@ def make_task(self):
raise NotImplementedError("This class is not supposed to supply make_task")


class FooTask(ExplorationTask):
def __init__(
self,
conf_name="conf.lmp",
conf_cont="",
inpu_name="in.lammps",
inpu_cont="",
):
super().__init__()
self._files = {
conf_name: conf_cont,
inpu_name: inpu_cont,
}


class FooTaskGroup(ExplorationTaskGroup):
def __init__(self, numb_task):
super().__init__()
Expand Down

0 comments on commit f281d1d

Please sign in to comment.