Skip to content

Commit

Permalink
Undo adding image collection to work unit. That needs refactoring som…
Browse files Browse the repository at this point in the history
…ewhere else.
  • Loading branch information
DinoBektesevic committed Jul 31, 2024
1 parent fef36b0 commit d02cf91
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/kbmod/image_collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -878,5 +878,5 @@ def toWorkUnit(self, search_config=None, **kwargs):
layeredImages.append(img)
imgstack = ImageStack(layeredImages)
if None not in self.wcs:
return WorkUnit(imgstack, search_config, per_image_wcs=list(self.wcs), collection=self)
return WorkUnit(imgstack, search_config, per_image_wcs=list(self.wcs))
return WorkUnit(imgstack, search_config)
4 changes: 3 additions & 1 deletion src/kbmod/work_unit.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,10 @@
import math
import os

from astropy.io import fits
from astropy.table import Table
from astropy.utils.exceptions import AstropyWarning
from astropy.wcs import WCS
from astropy.wcs.utils import skycoord_to_pixel
from astropy.time import Time
from astropy.coordinates import SkyCoord, EarthLocation
Expand All @@ -12,7 +15,6 @@

from kbmod.configuration import SearchConfiguration
from kbmod.search import ImageStack, LayeredImage, PSF, RawImage, Logging

from kbmod.wcs_utils import (
append_wcs_to_hdu_header,
extract_wcs_from_hdu_header,
Expand Down

0 comments on commit d02cf91

Please sign in to comment.