Skip to content

Commit

Permalink
proper function name
Browse files Browse the repository at this point in the history
  • Loading branch information
pecom committed Oct 6, 2023
1 parent 1967710 commit 3feb95e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions btk/blend_batch.py
Original file line number Diff line number Diff line change
Expand Up @@ -275,15 +275,15 @@ def __repr__(self):
return string

def save(self, path: str, batch_number: int = 0):
"""Save blend results into path."""
"""Save blend results into path using hdf5 format."""
for survey_name, blend_batch in self.results.items():
survey_directory = os.path.join(path, str(batch_number), survey_name)
if not os.path.exists(survey_directory):
os.makedirs(survey_directory)
blend_batch.save(survey_directory, batch_number)

def save(self, path: str, batch_number: int = 0):
"""Save blend results into path."""
def save_fits(self, path: str, batch_number: int = 0):
"""Save blend results into path using fits format."""
for survey_name, blend_batch in self.results.items():
survey_directory = os.path.join(path, str(batch_number), survey_name)
if not os.path.exists(survey_directory):
Expand Down

0 comments on commit 3feb95e

Please sign in to comment.