Fix a typo & add option to be faster

src/backscattering_experimental_data/factory/component.py

@@ -17,6 +17,7 @@ class ComponentFactory:
         start_time: Time,
         duration: Quantity[time],
         metadata: MetadataComponent,
+        force_virgotools: bool = False,
     ) -> Component:
         """
         Fetch a component from online data
@@ -31,11 +32,18 @@ class ComponentFactory:
             Duration
         metadata : MetadataComponent
             Component's metadata
+        force_virgotools : bool
+            Force the use of the virgotools library in place of gwpy
+            first to be faster. Default to False
+        Returns
+        -------
+        Component
         """
         sensor = download_timeseries(
             channel="V1:Sc_{}_MIR_Z".format(metadata.name),
             start_time=start_time,
             duration=duration,
             unit=uV,  # in um, but the calibration factor is considered as m/V
+            force_virgotools=force_virgotools,
         )
         return Component(metadata=metadata, sensor=sensor)

src/backscattering_experimental_data/factory/measurement.py

@@ -36,6 +36,7 @@ class MeasurementFactory:
         start_time: Time,
         duration: Quantity[time],
         initializer_metadata: list[InitializerScatterer],
+        force_virgotools: bool = False,
     ) -> Measurement:
         """
         Fetch a measurement from online data
@@ -55,6 +56,32 @@ class MeasurementFactory:
         initializer_metadata : list[InitializerScatterer]
             List of component's and scatterer's pre-initialized
             metadata that will be used in the created measurement
+        force_virgotools : bool
+            Force the use of the virgotools library in place of gwpy
+            first to be faster. Default to False
+        Examples
+        --------
+        >>> data = MeasurementFactory.fetch(
+        ...     name="a name",
+        ...     description="a detailed description",
+        ...     start_time=Time(
+        ...         "2025-01-01T00:00:00.000000000",
+        ...         format="isot",
+        ...         scale="utc",
+        ...     ),
+        ...     duration=10 * s,
+        ...     initializer_metadata=[
+        ...         InitializerScatterer(
+        ...             name="SDB1",
+        ...             calibration_factor=1.0 * m / V,
+        ...             associated_component=InitializerComponent(
+        ...                 name="SR",
+        ...                 calibration_factor=1.0 * m / V,
+        ...             ),
+        ...             injection=None,
+        ...         ),
+        ...     ],
+        ... )
         Returns
         -------
         Measurement : constructed Measurement from these metadata
@@ -70,6 +97,7 @@ class MeasurementFactory:
             duration=duration,
             units=list(channels_map.values()),
             frame="trend",
+            force_virgotools=force_virgotools,
         )
         if sum(data["V1:META_ITF_LOCK_index"].diff().abs()) != 0:
             raise Exception(
@@ -101,6 +129,7 @@ class MeasurementFactory:
                         name=initializer_scatterer.associated_component.name,
                         calibration_factor=initializer_scatterer.associated_component.calibration_factor,
                     ),
+                    force_virgotools=force_virgotools,
                 )
                 components.append(component)
             scatterers.append(
@@ -120,6 +149,7 @@ class MeasurementFactory:
             start_time=start_time,
             duration=duration,
             unit=None,
+            force_virgotools=force_virgotools,
         )
         return Measurement(
             metadata=metadata,

src/backscattering_experimental_data/factory/scatterer.py

@@ -17,6 +17,7 @@ class ScattererFactory:
         start_time: Time,
         duration: Quantity[time],
         metadata: MetadataScatterer,
+        force_virgotools: bool = False,
     ) -> Scatterer:
         """
         Fetch a component from online data
@@ -31,11 +32,18 @@ class ScattererFactory:
             Duration
         metadata : MetadataComponent
             Component's metadata
+        force_virgotools : bool
+            Force the use of the virgotools library in place of gwpy
+            first to be faster. Default to False
+        Returns
+        -------
+        Scatterer
         """
         sensor = download_timeseries(
             channel="V1:{}_LC_Z".format(metadata.name),
             start_time=start_time,
             duration=duration,
             unit=uV,  # in um, but the calibration factor is considered as m/V
+            force_virgotools=force_virgotools,
         )
         return Scatterer(metadata=metadata, sensor=sensor)

src/backscattering_experimental_data/utils.py

@@ -11,9 +11,9 @@ from gwpy.timeseries import TimeSeries, TimeSeriesDict
 try:
     from virgotools.frame_lib import FrameFile, FrVect  # pyright: ignore[reportMissingImports]
 
-    virgotool = True
+    virgotools = True
 except ImportError:
-    virgotool = False
+    virgotools = False
     pass
 
 
@@ -51,6 +51,7 @@ def download_timeseries(
     duration: Quantity[time],
     unit: Unit | None = None,
     frame: Literal["raw"] | Literal["trend"] = "raw",
+    force_virgotools: bool = False,
 ) -> TimeSeries:
     """
     Download a TimeSeries from an online channel
@@ -66,10 +67,14 @@ def download_timeseries(
         Unit of the TimeSeries (default ot None)
     frame : raw|trend
         Which frame to take the data from. Default to raw
+    force_virgotools : bool
+        Force the use of the virgotools library in place of gwpy first
+        to be faster. Default to False
     Notes
     -----
     It first try to use gwpy, and fallback to virgotools
     """
+    if not force_virgotools:
         try:
             return TimeSeries.get(
                 channel=channel,
@@ -77,11 +82,14 @@ def download_timeseries(
                 end=to_gps(start_time + duration),
             )
         except HTTPError:
-        if virgotool:
+            pass
+    if virgotools:
         vect: FrVect | None = None  # pyright: ignore[reportPossiblyUnboundVariable]
         with FrameFile(frame) as ffl:  # pyright: ignore[reportPossiblyUnboundVariable]
             vect = ffl.getChannel(  # pyright: ignore[reportPossiblyUnboundVariable]
-                    channel, to_gps(start_time), duration.to(s)
+                channel,
+                to_gps(start_time),
+                duration.to(s).value,
             )
         if vect is None:  # pyright: ignore[reportPossiblyUnboundVariable]
             raise RuntimeError(
@@ -105,6 +113,7 @@ def download_dicttimeseries(
     duration: Quantity[time],
     units: list[Unit | None] | None = None,
     frame: Literal["raw"] | Literal["trend"] = "raw",
+    force_virgotools: bool = False,
 ) -> TimeSeriesDict:
     """
     Download multiple TimeSeries at the same time to be more efficient
@@ -121,10 +130,14 @@ def download_dicttimeseries(
         channels. Default to None
     frame : raw|trend
         Which frame to take the data from. Default to raw
+    force_virgotools : bool
+        Force the use of the virgotools library in place of gwpy first
+        to be faster. Default to False
     Notes
     -----
     It first try to use gwpy, and fallback to virgotools
     """
+    if not force_virgotools:
         try:
             return TimeSeriesDict.get(
                 channels=channels,
@@ -132,13 +145,16 @@ def download_dicttimeseries(
                 end=to_gps(start_time + duration),
             )
         except HTTPError:
-        if virgotool:
+            pass
+    if virgotools:
         vects: list[FrVect] = []  # pyright: ignore[reportPossiblyUnboundVariable]
         with FrameFile(frame) as ffl:  # pyright: ignore[reportPossiblyUnboundVariable]
             for channel in channels:
                 vects.append(
                     ffl.getChannel(
-                            channel, to_gps(start_time), duration.to(s)
+                        channel,
+                        to_gps(start_time),
+                        duration.to(s).value,
                     )
                 )
         timeseries_dict: dict[str, TimeSeries] = dict()