Commit 11cc3a52 authored by Wei Chengliang's avatar Wei Chengliang
Browse files

fix: add None check for starDDL in free_mem

parent ff055172

catalog/newCatalog.py

+5 −2
Original line number Diff line number Diff line
@@ -118,6 +118,7 @@ class Catalog(CatalogBase): 
        self.pointing = pointing



        self.max_size = 0.0

        self.starDDL = None



        # [TODO] Milky Way extinction

        if (
@@ -695,8 +696,10 @@ class Catalog(CatalogBase): 
            self.objs.append(obj)



    def free_mem(self, **kward):

        if self.starDDL:

            self.starDDL.freeGlobeData()

            del self.starDDL

            self.starDDL = None



    def _load(self, **kwargs):

        self.objs = []

observation_sim/ObservationSim.py

+1 −1
Original line number Diff line number Diff line
@@ -210,7 +210,7 @@ class Observation(object): 
            except Exception as e:

                traceback.print_exc()

                chip_output.Log_error(e)

                chip_output.Log_error("Failed simulation on step: %s" % (step))

                chip_output.Log_error("Failed simulation on step: %s" % (step), stderr=True)

                break

        del chip.img

        del chip.flat_img