Commit 8d9ce0259afa06a38666631e747173ee781e42ca
1 parent
2961bc4c
Adapt to handling of DRM models (#3910)
Showing
2 changed files
with
2 additions
and
2 deletions
modules/comscripts/comlixfit.py
| ... | ... | @@ -120,7 +120,7 @@ class comlixfit(ctools.cslikelihood): |
| 120 | 120 | # Extract source models from observation model container |
| 121 | 121 | models = gammalib.GModels() |
| 122 | 122 | for model in self.obs().models(): |
| 123 | - if model.classname() == 'GModelSky': | |
| 123 | + if model.classname() == 'GModelSky' or model.classname() == 'GCOMModelDRM': | |
| 124 | 124 | models.append(model) |
| 125 | 125 | |
| 126 | 126 | # Loop over all observations | ... | ... |
modules/comscripts/comlixmap.py
| ... | ... | @@ -127,7 +127,7 @@ class comlixmap(ctools.cslikelihood): |
| 127 | 127 | # Extract source models from observation model container |
| 128 | 128 | models = gammalib.GModels() |
| 129 | 129 | for model in self.obs().models(): |
| 130 | - if model.classname() == 'GModelSky': | |
| 130 | + if model.classname() == 'GModelSky' or model.classname() == 'GCOMModelDRM': | |
| 131 | 131 | models.append(model) |
| 132 | 132 | |
| 133 | 133 | # Loop over all observations | ... | ... |