sparclclient 1.2.9b1__py2.py3-none-any.whl → 1.3.0b2__py2.py3-none-any.whl

sparcl/__init__.py

@@ -38,4 +38,4 @@ __all__ = ["client", "align_records"]
 #__version__ = "1.2.6"
 #__version__ = "1.2.7"
 #__version__ = "1.2.8"
-__version__ = "1.2.9b1"
+__version__ = "1.3.0b2"

sparcl/specutils.py

@@ -12,7 +12,32 @@ from astropy.nddata import InverseVariance
 import astropy.units as u
 
 def _validate_records(records, r0, collection):
-    """Validate that records can be converted to Spectrum."""
+    """Validate that records can be converted to Spectrum.
+
+    Parameters
+    ----------
+    records : list of dict
+        All records to validate.
+    r0 : dict
+        First record, used as reference for validation.
+    collection: bool
+        If ``True``, attempt to convert to a
+        :class:`~specutils.SpectrumCollection` instead.
+
+    Raises
+    ------
+    ValueError
+        If records lack 'wavelength' attribute.
+        If records have different data releases.
+        If wavelength array lengths differ (suggests using SpectrumList).
+        If wavelength pixel values differ and collection=False
+        (suggests using SpectrumCollection).
+
+    Warnings
+    --------
+    UserWarning
+        If records come from different data releases.
+    """
 
     # Check if the first record has wavelength data
     if 'wavelength' not in r0:
@@ -39,7 +64,45 @@ def _validate_records(records, r0, collection):
 def _extract_record_data(records, flux, uncertainty, mask, model, redshift,
                          meta, spectral_axis, has_model, has_redshift,
                          collection, single_record):
-    """Extract all data from records into arrays."""
+    """Extract all data from records into arrays. This function modifies
+    arrays in-place rather than returning values.
+
+    Parameters
+    ----------
+    records: list of dict
+        Records containing flux, ivar, mask, wavelength, and optional
+        model/redshift.
+    flux : np.ndarray
+        Pre-allocated array for flux values. Shape: (n_pixels,) if
+        single_record, else (n_records, n_pixels).
+    uncertainty : np.ndarray
+        Pre-allocated array for inverse variance.
+    mask : np.ndarray
+        Pre-allocated array for data quality masks.
+    model : np.ndarray or None
+        Pre-allocated array for model values if has_model=True.
+    redshift : list
+        Empty list to populate with redshift values.
+    meta : dict
+        Empty dict to populate with metadata.
+    spectral_axis : np.ndarray
+        For collections, pre-allocated 2D array to store wavelength grids.
+        For non-collections, this is just r0.wavelength (not modified).
+    has_model : bool
+        Whether records contain 'model' attribute.
+    has_redshift : bool
+        Whether records contain 'redshift' attribute.
+    collection : bool
+        If True, stores wavelength arrays for each record in spectral_axis.
+    single_record : bool
+        If True, treats arrays as 1D. If False, treats as 2D with row per
+        record.
+
+    Returns
+    -------
+    None
+        All outputs are written to the input arrays/containers in-place.
+    """
     for k, record in enumerate(records):
         if single_record:
             # For single record, assign directly (1D)
@@ -83,7 +146,7 @@ def to_Spectrum(results, *, collection=False):
     ----------
     results : :class:`sparcl.Results.Retrieved`
         Retrieved results, or a single record from a set of results.
-    collection : :class:`bool`, optional
+    collection : bool, optional
         If ``True``, attempt to convert to a
         :class:`~specutils.SpectrumCollection` instead.
 
@@ -131,16 +194,16 @@ def to_Spectrum(results, *, collection=False):
 
     # Build spectral axis
     if collection:
-        spectral_axis = np.zeros((len(records), r0.wavelength.shape[0]),
+        spectral_axis = np.empty((len(records), r0.wavelength.shape[0]),
                                  dtype=r0.wavelength.dtype)
     else:
         spectral_axis = r0.wavelength
 
     # Initialize arrays
-    flux = np.zeros(flux_shape, dtype=r0.flux.dtype)
-    uncertainty = np.zeros(flux_shape, dtype=r0.ivar.dtype)
-    mask = np.zeros(flux_shape, dtype=r0.mask.dtype)
-    model = np.zeros(flux_shape, dtype=r0.model.dtype) if has_model else None
+    flux = np.empty(flux_shape, dtype=r0.flux.dtype)
+    uncertainty = np.empty(flux_shape, dtype=r0.ivar.dtype)
+    mask = np.empty(flux_shape, dtype=r0.mask.dtype)
+    model = np.empty(flux_shape, dtype=r0.model.dtype) if has_model else None
     redshift = []
     meta = {}
 
@@ -152,6 +215,9 @@ def to_Spectrum(results, *, collection=False):
     # Convert redshift list to numpy array if exists
     if has_redshift:
         redshift = np.array(redshift)
+        if single_record and len(redshift) == 1:
+            # Convert to scalar if single record
+            redshift = redshift[0]
     else:
         redshift = None
 
@@ -161,7 +227,7 @@ def to_Spectrum(results, *, collection=False):
 
     # Prepare arguments common to both Spectrum and SpectrumCollection
     common_args = {
-        'flux': flux * u.Unit('1e-17 erg cm-2 s-1 AA-1'),
+        'flux': flux * u.Unit('10**-17 erg cm-2 s-1 AA-1'),
         'spectral_axis': spectral_axis * u.AA,
         'uncertainty': InverseVariance(uncertainty),
         'mask': mask,
@@ -200,7 +266,7 @@ def to_SpectrumList(results):
             if attribute not in ('flux', 'wavelength', 'ivar',
                                  'redshift', 'mask'):
                 meta[attribute] = r[attribute]
-        s1 = Spectrum(flux=r.flux*u.Unit('1e-17 erg cm-2 s-1 AA-1'),
+        s1 = Spectrum(flux=r.flux*u.Unit('10**-17 erg cm-2 s-1 AA-1'),
                       spectral_axis=r.wavelength*u.AA,
                       uncertainty=InverseVariance(r.ivar),
                       redshift=redshift,

{sparclclient-1.2.9b1.dist-info → sparclclient-1.3.0b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sparclclient
-Version: 1.2.9b1
+Version: 1.3.0b2
 Summary: A client for getting spectra and meta-data from NOIRLab.
 Author-email: "S. Pothier" <datalab-spectro@noirlab.edu>
 Description-Content-Type: text/markdown

{sparclclient-1.2.9b1.dist-info → sparclclient-1.3.0b2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sparcl/Results.py,sha256=IlnYZrDddARv3PJ3HPcyHJwTdMroqGdw-eaSrbmAA64,9290
-sparcl/__init__.py,sha256=YqaePEJGAUXC3_s-RCRYbHQlTaC_Bd6W9ObvBHvQXJ8,1193
+sparcl/__init__.py,sha256=akq-_0bDWSDTnmS3BZ827HTpiWKYYRO8qtlOpXd4ax8,1193
 sparcl/client.py,sha256=OAY9F1KbyVc5kos7twqOFnQcGtoqraNG6-CJWqmVE3o,39976
 sparcl/conf.py,sha256=GFNDelaiVIAkjNjvFlG7HAlPpU39nqZmTPohQGmOcgI,928
 sparcl/exceptions.py,sha256=sznmOMGENHvxutSXlRVWqi87bR2Qiebka7LyR3mAII0,4244
@@ -7,7 +7,7 @@ sparcl/fields.py,sha256=NZUBqDidpbXfeX5F4b306F323xZY2CRIx8eVv-HWTVU,5127
 sparcl/gather_2d.py,sha256=YTFVQl38dRZjUu0AEUFVZtQm7zC8mU3LVvUTDGxp6u8,8722
 sparcl/resample_spectra.py,sha256=nk5HiyaGF-b9SiTmIC9yJBbYX4VYsiLFHyW186bPs2s,1310
 sparcl/sparc.ini,sha256=q_wjo9DLnCYRxWFMl0CtMYp4DD1AXfEcK6BP6cmncwo,329
-sparcl/specutils.py,sha256=IKI4VQ53_ay6kgfAGdvNkpseCH0eURes3RupGJ2-JVA,8738
+sparcl/specutils.py,sha256=QUwlx0bsxEdG50WQ5Fp3uLvDsoDKg5Zl5Te-xnIVRDs,10968
 sparcl/type_conversion.py,sha256=QmXNX9j_7QHnBu83f2ZBfREoql9wuo98ZbhQtSjRRWc,12965
 sparcl/unsupported.py,sha256=bfkkZa-PuqwN-Bqo3vCIrLupbWMDTCiTHPMNfXnqmMc,1848
 sparcl/utils.py,sha256=pDAk9roe7ezfVohnKcLMxFjjXHkp7EQLbgVNe5sSTrk,6259
@@ -16,7 +16,7 @@ sparcl/benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 sparcl/benchmarks/benchmarks.py,sha256=OmlSdnAPLmcvGXsr-HzGyfAAcnoqlO0JQ4EIA7JGhZc,9424
 sparcl/benchmarks/sparcl_benchmarking.ipynb,sha256=gwof2hqM9Qb49qzRX-mky7WNqXZCMSB7ry8bX8dImxc,17559
 sparcl/notebooks/sparcl-examples.ipynb,sha256=gEwMKI1x7A1YsVeCsQn1QoMO0ZuIhMUAu3qedTiQ7hM,169268
-sparclclient-1.2.9b1.dist-info/LICENSE,sha256=y10EluGMCzGs9X4oYCYyix3l6u-lawB_vlGR8qe442Q,1576
-sparclclient-1.2.9b1.dist-info/WHEEL,sha256=ssQ84EZ5gH1pCOujd3iW7HClo_O_aDaClUbX4B8bjKY,100
-sparclclient-1.2.9b1.dist-info/METADATA,sha256=0L79znAjI_0CMKc_AUJgjebyNgf1mdrvFHOQ7tpi20o,677
-sparclclient-1.2.9b1.dist-info/RECORD,,
+sparclclient-1.3.0b2.dist-info/LICENSE,sha256=y10EluGMCzGs9X4oYCYyix3l6u-lawB_vlGR8qe442Q,1576
+sparclclient-1.3.0b2.dist-info/WHEEL,sha256=ssQ84EZ5gH1pCOujd3iW7HClo_O_aDaClUbX4B8bjKY,100
+sparclclient-1.3.0b2.dist-info/METADATA,sha256=_fYIDfWnBvAOom0dDYuB1G_5zK3WxNXDzEAJYyNpKMg,677
+sparclclient-1.3.0b2.dist-info/RECORD,,