Coverage for python/lsst/ctrl/mpexec/taskFactory.py: 30%
22 statements
« prev ^ index » next coverage.py v7.3.2, created at 2023-12-03 10:43 +0000
« prev ^ index » next coverage.py v7.3.2, created at 2023-12-03 10:43 +0000
1# This file is part of ctrl_mpexec.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (http://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This software is dual licensed under the GNU General Public License and also
10# under a 3-clause BSD license. Recipients may choose which of these licenses
11# to use; please see the files gpl-3.0.txt and/or bsd_license.txt,
12# respectively. If you choose the GPL option then the following text applies
13# (but note that there is still no warranty even if you opt for BSD instead):
14#
15# This program is free software: you can redistribute it and/or modify
16# it under the terms of the GNU General Public License as published by
17# the Free Software Foundation, either version 3 of the License, or
18# (at your option) any later version.
19#
20# This program is distributed in the hope that it will be useful,
21# but WITHOUT ANY WARRANTY; without even the implied warranty of
22# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
23# GNU General Public License for more details.
24#
25# You should have received a copy of the GNU General Public License
26# along with this program. If not, see <http://www.gnu.org/licenses/>.
28from __future__ import annotations
30__all__ = ["TaskFactory"]
32import logging
33from collections.abc import Iterable
34from typing import TYPE_CHECKING, Any
36from lsst.pipe.base import TaskFactory as BaseTaskFactory
38if TYPE_CHECKING:
39 from lsst.daf.butler import DatasetRef, LimitedButler
40 from lsst.pipe.base import PipelineTask, TaskDef
42_LOG = logging.getLogger(__name__)
45class TaskFactory(BaseTaskFactory):
46 """Class instantiating PipelineTasks."""
48 def makeTask(
49 self, taskDef: TaskDef, butler: LimitedButler, initInputRefs: Iterable[DatasetRef] | None
50 ) -> PipelineTask:
51 # docstring inherited
53 config = taskDef.config
55 # Get init inputs from butler.
56 init_inputs: dict[str, Any] = {}
57 if initInputRefs:
58 connections = config.connections.ConnectionsClass(config=config)
59 for name in connections.initInputs:
60 attribute = getattr(connections, name)
61 dataset_type_name = attribute.name
62 for ref in initInputRefs:
63 if ref.datasetType.name == dataset_type_name:
64 init_inputs[name] = butler.get(ref)
65 break
67 # make task instance
68 task = taskDef.taskClass(config=config, initInputs=init_inputs, name=taskDef.label)
69 return task