astro-otter 0.0.1__tar.gz → 0.0.2__tar.gz

{astro_otter-0.0.1/src/astro_otter.egg-info → astro_otter-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: astro-otter
-Version: 0.0.1
+Version: 0.0.2
 Author-email: Noah Franz <nfranz@arizona.edu>
 License: MIT License
         
@@ -50,6 +50,13 @@ Requires-Dist: synphot
 Requires-Dist: ads
 Requires-Dist: ruff
 Requires-Dist: pre-commit
+Requires-Dist: Sphinx>=3.0.0
+Requires-Dist: myst_parser>=0.13
+Requires-Dist: nbsphinx>=0.9.1
+Requires-Dist: sphinx-book-theme>=0.0.33
+Requires-Dist: sphinx_copybutton
+Requires-Dist: autodoc
+Requires-Dist: ipykernel
 
 # OTTER API
 ### **O**pen mul**T**iwavelength **T**ransient **E**vent **R**epository
@@ -66,9 +73,9 @@ A Python API for the OTTER.
 [github-discussions-link]:  https://github.com/mattbellis/hepfile/discussions
 [gitter-badge]:             https://badges.gitter.im/https://github.com/mattbellis/hepfile/community.svg
 [gitter-link]:              https://gitter.im/https://github.com/mattbellis/hepfile/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge
-[pypi-link]:                https://pypi.org/project/hepfile/
-[pypi-platforms]:           https://img.shields.io/pypi/pyversions/hepfile
-[pypi-version]:             https://badge.fury.io/py/hepfile.svg
+[pypi-link]:                https://pypi.org/project/astro-otter/
+[pypi-platforms]:           https://img.shields.io/pypi/pyversions/astro-otter
+[pypi-version]:             https://badge.fury.io/py/astro-otter.svg
 [rtd-badge]:                https://readthedocs.org/projects/otter/badge/?version=latest
 [rtd-link]:                 https://otter.readthedocs.io/en/latest/?badge=latest
 [sk-badge]:                 https://scikit-hep.org/assets/images/Scikit--HEP-Project-blue.svg
@@ -77,7 +84,9 @@ A Python API for the OTTER.
 [codecov-badge]:            https://codecov.io/gh/astro-otter/otter/graph/badge.svg?token=BtCerOdTc0
 [codecov-link]:             https://codecov.io/gh/astro-otter/otter
 
+[![Documentation Status](https://readthedocs.org/projects/astro-otter/badge/?version=latest)](https://astro-otter.readthedocs.io/en/latest/?badge=latest)
 [![Actions Status][actions-badge]][actions-link]
+[![PyPI version][pypi-version]][pypi-link]
 [![Linting: Ruff][ruff-badge]][ruff-link]
 [![codecov][codecov-badge]][codecov-link]
 

{astro_otter-0.0.1 → astro_otter-0.0.2}/README.md

@@ -13,9 +13,9 @@ A Python API for the OTTER.
 [github-discussions-link]:  https://github.com/mattbellis/hepfile/discussions
 [gitter-badge]:             https://badges.gitter.im/https://github.com/mattbellis/hepfile/community.svg
 [gitter-link]:              https://gitter.im/https://github.com/mattbellis/hepfile/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge
-[pypi-link]:                https://pypi.org/project/hepfile/
-[pypi-platforms]:           https://img.shields.io/pypi/pyversions/hepfile
-[pypi-version]:             https://badge.fury.io/py/hepfile.svg
+[pypi-link]:                https://pypi.org/project/astro-otter/
+[pypi-platforms]:           https://img.shields.io/pypi/pyversions/astro-otter
+[pypi-version]:             https://badge.fury.io/py/astro-otter.svg
 [rtd-badge]:                https://readthedocs.org/projects/otter/badge/?version=latest
 [rtd-link]:                 https://otter.readthedocs.io/en/latest/?badge=latest
 [sk-badge]:                 https://scikit-hep.org/assets/images/Scikit--HEP-Project-blue.svg
@@ -24,7 +24,9 @@ A Python API for the OTTER.
 [codecov-badge]:            https://codecov.io/gh/astro-otter/otter/graph/badge.svg?token=BtCerOdTc0
 [codecov-link]:             https://codecov.io/gh/astro-otter/otter
 
+[![Documentation Status](https://readthedocs.org/projects/astro-otter/badge/?version=latest)](https://astro-otter.readthedocs.io/en/latest/?badge=latest)
 [![Actions Status][actions-badge]][actions-link]
+[![PyPI version][pypi-version]][pypi-link]
 [![Linting: Ruff][ruff-badge]][ruff-link]
 [![codecov][codecov-badge]][codecov-link]
 

{astro_otter-0.0.1 → astro_otter-0.0.2}/pyproject.toml

@@ -39,8 +39,17 @@ dependencies = [
 
   # useful dev tools
   "ruff", # linter
-  "pre-commit" # enable pre-commit hooks
-  ]
+  "pre-commit", # enable pre-commit hooks
+
+  # tools for building the readthedocs page
+  "Sphinx>=3.0.0",
+  "myst_parser>=0.13",
+  "nbsphinx>=0.9.1",
+  "sphinx-book-theme>=0.0.33",
+  "sphinx_copybutton",
+  "autodoc",
+  "ipykernel"
+]
 
 [project.urls]
 Home = "https://github.com/astro-otter"

{astro_otter-0.0.1 → astro_otter-0.0.2/src/astro_otter.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: astro-otter
-Version: 0.0.1
+Version: 0.0.2
 Author-email: Noah Franz <nfranz@arizona.edu>
 License: MIT License
         
@@ -50,6 +50,13 @@ Requires-Dist: synphot
 Requires-Dist: ads
 Requires-Dist: ruff
 Requires-Dist: pre-commit
+Requires-Dist: Sphinx>=3.0.0
+Requires-Dist: myst_parser>=0.13
+Requires-Dist: nbsphinx>=0.9.1
+Requires-Dist: sphinx-book-theme>=0.0.33
+Requires-Dist: sphinx_copybutton
+Requires-Dist: autodoc
+Requires-Dist: ipykernel
 
 # OTTER API
 ### **O**pen mul**T**iwavelength **T**ransient **E**vent **R**epository
@@ -66,9 +73,9 @@ A Python API for the OTTER.
 [github-discussions-link]:  https://github.com/mattbellis/hepfile/discussions
 [gitter-badge]:             https://badges.gitter.im/https://github.com/mattbellis/hepfile/community.svg
 [gitter-link]:              https://gitter.im/https://github.com/mattbellis/hepfile/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge
-[pypi-link]:                https://pypi.org/project/hepfile/
-[pypi-platforms]:           https://img.shields.io/pypi/pyversions/hepfile
-[pypi-version]:             https://badge.fury.io/py/hepfile.svg
+[pypi-link]:                https://pypi.org/project/astro-otter/
+[pypi-platforms]:           https://img.shields.io/pypi/pyversions/astro-otter
+[pypi-version]:             https://badge.fury.io/py/astro-otter.svg
 [rtd-badge]:                https://readthedocs.org/projects/otter/badge/?version=latest
 [rtd-link]:                 https://otter.readthedocs.io/en/latest/?badge=latest
 [sk-badge]:                 https://scikit-hep.org/assets/images/Scikit--HEP-Project-blue.svg
@@ -77,7 +84,9 @@ A Python API for the OTTER.
 [codecov-badge]:            https://codecov.io/gh/astro-otter/otter/graph/badge.svg?token=BtCerOdTc0
 [codecov-link]:             https://codecov.io/gh/astro-otter/otter
 
+[![Documentation Status](https://readthedocs.org/projects/astro-otter/badge/?version=latest)](https://astro-otter.readthedocs.io/en/latest/?badge=latest)
 [![Actions Status][actions-badge]][actions-link]
+[![PyPI version][pypi-version]][pypi-link]
 [![Linting: Ruff][ruff-badge]][ruff-link]
 [![codecov][codecov-badge]][codecov-link]
 

{astro_otter-0.0.1 → astro_otter-0.0.2}/src/astro_otter.egg-info/SOURCES.txt

@@ -16,5 +16,8 @@ src/otter/io/transient.py
 src/otter/plotter/__init__.py
 src/otter/plotter/otter_plotter.py
 src/otter/plotter/plotter.py
+tests/test_exceptions.py
+tests/test_otter.py
 tests/test_package.py
-tests/test_transient.py
+tests/test_transient.py
+tests/test_util.py

astro_otter-0.0.2/src/astro_otter.egg-info/requires.txt

@@ -0,0 +1,17 @@
+numpy>=1.20
+astropy>=5.2
+pandas
+matplotlib
+plotly
+astroquery
+synphot
+ads
+ruff
+pre-commit
+Sphinx>=3.0.0
+myst_parser>=0.13
+nbsphinx>=0.9.1
+sphinx-book-theme>=0.0.33
+sphinx_copybutton
+autodoc
+ipykernel

{astro_otter-0.0.1 → astro_otter-0.0.2}/src/otter/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 # get the version
 from ._version import __version__
 

{astro_otter-0.0.1 → astro_otter-0.0.2}/src/otter/_version.py

@@ -2,4 +2,4 @@
 Just define the package version in one place
 """
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"

{astro_otter-0.0.1 → astro_otter-0.0.2}/src/otter/exceptions.py

@@ -2,8 +2,14 @@
 Custom exceptions for otter
 """
 
+from __future__ import annotations
+
 
 class FailedQueryError(ValueError):
+    """
+    Exception thrown when the users query does not return any results.
+    """
+
     def __str__(self):
         txt = "You're query/search did not return any results! "
         txt += "Try again with different parameters!"
@@ -11,10 +17,19 @@ class FailedQueryError(ValueError):
 
 
 class IOError(ValueError):
+    """
+    Exception thrown when the input or output argument/value is not the correct type.
+    """
+
     pass
 
 
 class OtterLimitationError(Exception):
+    """
+    Exception thrown when the user requests something that is currently not supported
+    by the API.
+    """
+
     def __init__(self, msg):
         self.msg = "Current Limitation Found: " + msg
 
@@ -23,4 +38,8 @@ class OtterLimitationError(Exception):
 
 
 class TransientMergeError(Exception):
+    """
+    Exception thrown when the Transient objects can not be combined as expected.
+    """
+
     pass

{astro_otter-0.0.1 → astro_otter-0.0.2}/src/otter/io/otter.py

@@ -2,6 +2,7 @@
 This is the primary class for user interaction with the catalog
 """
 
+from __future__ import annotations
 import os
 import json
 import glob
@@ -28,14 +29,11 @@ class Otter(object):
     This is the primary class for users to access the otter backend database
 
     Args:
-        username [str]: Your connection username to the database, default is the user
-                        login which only has read permission.
-        password [str]: Your password corresponding to your username.
-        db [str]: The database name to connect to. This is default to 'otter' which is
-                  the only database so far.
-        collection [str]: The collection to read data from. Right now the only
-                          collection is 'tdes'.
-        debug [bool]: debug mode, set to true to limit reading from database.
+        datadir (str): Path to the data directory with the otter data. If not provided
+                       will default to a ".otter" directory in the CWD where you call
+                       this class from.
+        debug (bool): If we should just debug and not do anything serious.
+
     """
 
     def __init__(self, datadir: str = None, debug: bool = False) -> None:
@@ -87,9 +85,9 @@ class Otter(object):
         Performs a cone search of the catalog over the given coords and radius.
 
         Args:
-            coords [SkyCoord]: An astropy SkyCoord object with coordinates to match to
-            radius [float]: The radius of the cone in arcseconds, default is 0.05"
-            raw [bool]: If False (the default) return an astropy table of the metadata
+            coords (SkyCoord): An astropy SkyCoord object with coordinates to match to
+            radius (float): The radius of the cone in arcseconds, default is 0.05"
+            raw (bool): If False (the default) return an astropy table of the metadata
                         for matching objects. Otherwise, return the raw json dicts
 
         Return:
@@ -117,35 +115,40 @@ class Otter(object):
         unit conversion for you!
 
         Args:
-            flux_units [astropy.unit.Unit]: Either a valid string to convert
+            flux_units (astropy.unit.Unit): Either a valid string to convert
                                             or an astropy.unit.Unit
-            date_units [astropy.unit.Unit]: Either a valid string to convert to a date
+            date_units (astropy.unit.Unit): Either a valid string to convert to a date
                                             or an astropy.unit.Unit
-            return_type [str]: Either 'astropy' or 'pandas'. If astropy, returns an
+            return_type (str): Either 'astropy' or 'pandas'. If astropy, returns an
                                astropy Table. If pandas, returns a pandas DataFrame.
                                Default is 'astropy'.
-            obs_type [str]: Either 'radio', 'uvoir', or 'xray'. Will only return that
+            obs_type (str): Either 'radio', 'uvoir', or 'xray'. Will only return that
                             type of photometry if not None. Default is None and will
                             return any type of photometry.
-            keep_raw [bool]: If True, keep the raw flux/date/freq/wave associated with
+            keep_raw (bool): If True, keep the raw flux/date/freq/wave associated with
                              the dataset. Else, just keep the converted data. Default
                              is False.
-            **kwargs : Arguments to pass to Otter.query(). Can be:
-                       names [list[str]]: A list of names to get the metadata for
-                       coords [SkyCoord]: An astropy SkyCoord object with coordinates
+            **kwargs : Arguments to pass to Otter.query(). Can be::
+
+                       names (list[str]): A list of names to get the metadata for
+                       coords (SkyCoord): An astropy SkyCoord object with coordinates
                                           to match to
-                       radius [float]: The radius in arcseconds for a cone search,
+                       radius (float): The radius in arcseconds for a cone search,
                                        default is 0.05"
-                       minZ [float]: The minimum redshift to search for
-                       maxZ [float]: The maximum redshift to search for
-                       refs [list[str]]: A list of ads bibcodes to match to. Will only
+                       minZ (float): The minimum redshift to search for
+                       maxZ (float): The maximum redshift to search for
+                       refs (list[str]): A list of ads bibcodes to match to. Will only
                                          return metadata for transients that have this
                                          as a reference.
-                       hasSpec [bool]: if True, only return events that have spectra.
+                       hasSpec (bool): if True, only return events that have spectra.
 
         Return:
-           The photometry for the requested transients that match the arguments.
-           Will be an astropy Table sorted by transient default name.
+            The photometry for the requested transients that match the arguments.
+            Will be an astropy Table sorted by transient default name.
+
+        Raises:
+            FailedQueryError: When the query returns no results
+            IOError: if one of your inputs is incorrect
         """
         queryres = self.query(hasphot=True, **kwargs)
 
@@ -153,16 +156,25 @@ class Otter(object):
         for transient in queryres:
             # clean the photometry
             default_name = transient["name/default_name"]
-            phot = transient.clean_photometry(
-                flux_unit=flux_unit,
-                date_unit=date_unit,
-                wave_unit=wave_unit,
-                freq_unit=freq_unit,
-                obs_type=obs_type,
-            )
-            phot["name"] = [default_name] * len(phot)
 
-            dicts.append(phot)
+            try:
+                phot = transient.clean_photometry(
+                    flux_unit=flux_unit,
+                    date_unit=date_unit,
+                    wave_unit=wave_unit,
+                    freq_unit=freq_unit,
+                    obs_type=obs_type,
+                )
+
+                phot["name"] = [default_name] * len(phot)
+
+                dicts.append(phot)
+
+            except FailedQueryError:
+                # This is fine, it just means that there is no data associated
+                # with this one transient. We'll check and make sure there is data
+                # associated with at least one of the transients later!
+                pass
 
         if len(dicts) == 0:
             raise FailedQueryError()
@@ -203,7 +215,7 @@ class Otter(object):
         Loads an otter JSON file
 
         Args:
-            filename [str]: The path to the file to load
+            filename (str): The path to the OTTER JSON file to load
         """
 
         # read in files from summary
@@ -234,15 +246,15 @@ class Otter(object):
         same units.
 
         Args:
-            names [list[str]]: A list of names to get the metadata for
-            coords [SkyCoord]: An astropy SkyCoord object with coordinates to match to
-            radius [float]: The radius in arcseconds for a cone search, default is 0.05"
-            minz [float]: The minimum redshift to search for
-            maxz [float]: The maximum redshift to search for
-            refs [list[str]]: A list of ads bibcodes to match to. Will only return
+            names (list[str]): A list of names to get the metadata for
+            coords (SkyCoord): An astropy SkyCoord object with coordinates to match to
+            radius (float): The radius in arcseconds for a cone search, default is 0.05"
+            minz (float): The minimum redshift to search for
+            maxz (float): The maximum redshift to search for
+            refs (list[str]): A list of ads bibcodes to match to. Will only return
                               metadata for transients that have this as a reference.
-            hasphot [bool]: if True, only returns transients which have photometry.
-            hasspec [bool]: if True, only return transients that have spectra.
+            hasphot (bool): if True, only returns transients which have photometry.
+            hasspec (bool): if True, only return transients that have spectra.
 
         Return:
            Get all of the raw (unconverted!) data for objects that match the criteria.
@@ -344,24 +356,29 @@ class Otter(object):
 
         return outdata
 
-    def save(self, schema: list[dict], testing=False, **kwargs) -> None:
+    def save(self, schema: list[dict], testing=False) -> None:
         """
         Upload all the data in the given list of schemas.
 
         Args:
-            schema [list[dict]]: A list of json dictionaries
+            schema (list[dict]): A list of json dictionaries
+            testing (bool): Should we just enter test mode? Default is False
+
+        Raises:
+            OtterLimitationError: If some objects in OTTER are within 5" we can't figure
+                                  out which ones to merge with which ones.
         """
 
         if not isinstance(schema, list):
             schema = [schema]
 
         for transient in schema:
-            print(transient["name/default_name"])
-
             # convert the json to a Transient
             if not isinstance(transient, Transient):
                 transient = Transient(transient)
 
+            print(transient["name/default_name"])
+
             coord = transient.get_skycoord()
             res = self.cone_search(coords=coord)
 
@@ -405,8 +422,6 @@ class Otter(object):
             outfilepath = os.path.join(self.DATADIR, todel[0] + ".json")
             if test_mode:
                 print("Renaming the following file for backups: ", outfilepath)
-            else:
-                os.rename(outfilepath, outfilepath + ".backup")
         else:
             if test_mode:
                 print("Don't need to mess with the files at all!")
@@ -429,13 +444,16 @@ class Otter(object):
             print(f"Would write to {outfilepath}")
             # print(out)
 
-    def generate_summary_table(self, save=False):
+    def generate_summary_table(self, save=False) -> pd.DataFrame:
         """
         Generate a summary table for the JSON files in self.DATADIR
 
         args:
-            save [bool]: if True, save the summary file to "summary.csv"
-                         in self.DATADIR. Default is False.
+            save (bool): if True, save the summary file to "summary.csv"
+                         in self.DATADIR. Default is False and is just returned.
+
+        returns:
+            pandas.DataFrame of the summary meta information of the transients
         """
         allfiles = glob.glob(os.path.join(self.DATADIR, "*.json"))
 

{astro_otter-0.0.1 → astro_otter-0.0.2}/src/otter/io/transient.py

@@ -3,10 +3,12 @@ Class for a transient,
 basically just inherits the dict properties with some overwriting
 """
 
+from __future__ import annotations
 import warnings
 from copy import deepcopy
 import re
 from collections.abc import MutableMapping
+from typing_extensions import Self
 
 import numpy as np
 import pandas as pd
@@ -37,7 +39,9 @@ class Transient(MutableMapping):
         Overwrite the dictionary init
 
         Args:
-            d [dict]: A transient dictionary
+            d (dict): A transient dictionary
+            name (str): The default name of the transient, default is None and it will
+                        be inferred from the input dictionary.
         """
         self.data = d
 
@@ -68,7 +72,7 @@ class Transient(MutableMapping):
         """
 
         if isinstance(keys, (list, tuple)):
-            return Transient({key: self[key] for key in keys})
+            return Transient({key: (self[key] if key in self else []) for key in keys})
         elif isinstance(keys, str) and "/" in keys:  # this is for a path
             s = "']['".join(keys.split("/"))
             s = "['" + s
@@ -85,6 +89,10 @@ class Transient(MutableMapping):
             return self.data[keys]
 
     def __setitem__(self, key, value):
+        """
+        Override set item to work with the '/' syntax
+        """
+
         if isinstance(key, str) and "/" in key:  # this is for a path
             s = "']['".join(key.split("/"))
             s = "['" + s
@@ -109,7 +117,7 @@ class Transient(MutableMapping):
 
     def __repr__(self, html=False):
         if not html:
-            return str(self.data)
+            return f"Transient(\n\tName: {self.default_name},\n\tKeys: {self.keys()}\n)"
         else:
             html = ""
 
@@ -260,14 +268,17 @@ class Transient(MutableMapping):
         # now return out as a Transient Object
         return Transient(out)
 
-    def get_meta(self, keys=None):
+    def get_meta(self, keys=None) -> Self:
         """
         Get the metadata (no photometry or spectra)
 
         This essentially just wraps on __getitem__ but with some checks
 
         Args:
-            keys [list[str]] : list of keys
+            keys (list[str]) : list of keys to get the metadata for from the transient
+
+        Returns:
+            A Transient object of just the meta data
         """
         if keys is None:
             keys = list(self.keys())
@@ -296,9 +307,16 @@ class Transient(MutableMapping):
 
         return self[keys]
 
-    def get_skycoord(self, coord_format="icrs"):
+    def get_skycoord(self, coord_format="icrs") -> SkyCoord:
         """
         Convert the coordinates to an astropy SkyCoord
+
+        Args:
+            coord_format (str): Astropy coordinate format to convert the SkyCoord to
+                                defaults to icrs.
+
+        Returns:
+            Astropy.coordinates.SkyCoord of the default coordinate for the transient
         """
 
         # now we can generate the SkyCoord
@@ -309,9 +327,12 @@ class Transient(MutableMapping):
 
         return coord
 
-    def get_discovery_date(self):
+    def get_discovery_date(self) -> Time:
         """
-        Get the default discovery date
+        Get the default discovery date for this Transient
+
+        Returns:
+            astropy.time.Time of the default discovery date
         """
         key = "date_reference"
         date = self._get_default(key, filt='df["date_type"] == "discovery"')
@@ -322,9 +343,12 @@ class Transient(MutableMapping):
 
         return Time(date["value"], format=f)
 
-    def get_redshift(self):
+    def get_redshift(self) -> float:
         """
-        Get the default redshift
+        Get the default redshift of this Transient
+
+        Returns:
+            Float value of the default redshift
         """
         f = "df['distance_type']=='redshift'"
         default = self._get_default("distance", filt=f)
@@ -390,10 +414,32 @@ class Transient(MutableMapping):
         wave_unit: u.Unit = "nm",
         by: str = "raw",
         obs_type: str = None,
-    ):
+    ) -> pd.DataFrame:
         """
         Ensure the photometry associated with this transient is all in the same
         units/system/etc
+
+        Args:
+            flux_unit (astropy.unit.Unit): The astropy unit or string representation of
+                                           an astropy unit to convert and return the
+                                           flux as.
+            date_unit (str): Valid astropy date format string.
+            freq_unit (astropy.unit.Unit): The astropy unit or string representation of
+                                           an astropy unit to convert and return the
+                                           frequency as.
+            wave_unit (astropy.unit.Unit): The astropy unit or string representation of
+                                           an astropy unit to convert and return the
+                                           wavelength as.
+            by (str): Either 'raw' or 'value'. 'raw' is the default and is highly
+                      recommended! If 'value' is used it may skip some photometry.
+                      See the schema definition to understand this keyword completely
+                      before using it.
+            obs_type (str): "radio", "xray", or "uvoir". If provided, it only returns
+                            data taken within that range of wavelengths/frequencies.
+                            Default is None which will return all of the data.
+
+        Returns:
+            A pandas DataFrame of the cleaned up photometry in the requested units
         """
 
         # check inputs
@@ -429,6 +475,9 @@ class Transient(MutableMapping):
             else:
                 by = "value"
 
+        # skip rows where 'by' is nan
+        df = df[df[by].notna()]
+
         # drop irrelevant obs_types before continuing
         if obs_type is not None:
             valid_obs_types = {"radio", "uvoir", "xray"}

{astro_otter-0.0.1 → astro_otter-0.0.2}/src/otter/plotter/otter_plotter.py

@@ -6,19 +6,21 @@ Currently supported backends are:
 - plotly
 """
 
+from __future__ import annotations
 import importlib
 
 
 class OtterPlotter:
-    def __init__(self, backend):
-        """
-        Handles the backend for the "plotter" module
+    """
+    Handles the backend for the "plotter" module
 
-        Args:
-            backend [string]: a string of the module name to import and use
-                              as the backend. Currently supported are "matplotlib",
-                              "matplotlib.pyplot", "plotly", and "plotly.graph_objects"
-        """
+    Args:
+        backend (string): a string of the module name to import and use
+                          as the backend. Currently supported are "matplotlib",
+                          "matplotlib.pyplot", "plotly", and "plotly.graph_objects"
+    """
+
+    def __init__(self, backend):
         if backend == "matplotlib.pyplot":
             self.backend = backend
         elif backend == "pyplot.graph_objects":