Skip to content

Commit 87d14b7

Browse files
authored
Updated docs API reference (#99)
* docs: Added docstrings to file_io module * docs: added class docstrings to ModelFileRegistry. * docs: added method docstring to CatalogRepository.filter_catalog
1 parent 640c8b2 commit 87d14b7

File tree

4 files changed

+250
-14
lines changed

4 files changed

+250
-14
lines changed

docs/reference/api_reference.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,22 +150,27 @@ A small wrapper for ``pyCSEP`` readers
150150
.. autosummary::
151151
:nosignatures:
152152

153+
CatalogParser
153154
CatalogParser.ascii
154155
CatalogParser.json
155156

157+
CatalogSerializer
156158
CatalogSerializer.ascii
157159
CatalogSerializer.json
158160

161+
GriddedForecastParsers
159162
GriddedForecastParsers.dat
160163
GriddedForecastParsers.xml
161164
GriddedForecastParsers.quadtree
162165
GriddedForecastParsers.csv
163166
GriddedForecastParsers.hdf5
164167

168+
CatalogForecastParsers
165169
CatalogForecastParsers.csv
166170
CatalogForecastParsers.load_hermes_catalog
167171

168172
HDF5Serializer.grid2hdf5
173+
check_format
169174
serialize
170175

171176

floatcsep/infrastructure/registries.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,14 @@ def factory(
167167

168168

169169
class ModelFileRegistry(ModelRegistry, FilepathMixin):
170+
"""
171+
The class is responsible to handle the keys (in this case filepaths) to access model objects
172+
such as forecasts, input catalogs or argument/parameter files. These keys are used mainly by
173+
:class:`floatcsep.infrastructure.repositories.ForecastRepository` or
174+
:class:`floatcsep.infrastructure.repositories.CatalogRepository`.
175+
176+
"""
177+
170178
def __init__(
171179
self,
172180
model_name: str,

floatcsep/infrastructure/repositories.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,22 @@ def filter_catalog(
238238
max_depth=None,
239239
region=None,
240240
) -> CSEPCatalog:
241+
"""
242+
Wrapper for pyCSEP catalog filters, to constrain a catalog to a given time and magnitude
243+
range, as well to a spatial region.
244+
245+
Args:
246+
start_date (datetime.datetime): Initial datetime
247+
end_date (datetime.datetime): Final datetime
248+
min_mag (float): Minimum magnitude to filter
249+
max_mag (float): Maximum magnitude to filter
250+
min_depth (float): Minimum depth to filter (positive downwards from surface)
251+
max_depth (float): Maximum depth to filter (positive downwards from surface)
252+
region (csep.core.regions.CartesianGrid2D): Spatial domain to filter
241253
254+
Returns:
255+
Filtered catalog
256+
"""
242257
filters = []
243258
if start_date:
244259
filters.append(

0 commit comments

Comments
 (0)