27471184649

Committed 13 Jun 2026 03:37PM UTC coverage: 0.0% (-79.4%) from 79.431%

Build # 27471184649

Build Type

push

github

Committed by

MatthewGerber

Commit Message

* Change machine type.
* Try xdist.

Coverage Stats

0 of 5484 relevant lines covered (0.0%)

0.0 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/src/rlai/state_value/function_approximation/models/__init__.py

from abc import ABC
from typing import Optional

from matplotlib import pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages

from rlai.models import FunctionApproximationModel


class StateFunctionApproximationModel(FunctionApproximationModel, ABC):
    """
    Base class for models that approximate state-action value functions.
    """

    def plot(
            self,
            render: bool,
            pdf: Optional[PdfPages]
    ) -> Optional[plt.Figure]:
        """
        Plot the model. If called from the main thread and render is True, then a new plot will be generated. If called
        from a background thread, then the data used by the plot will be updated but a plot will not be generated or
        updated. This supports a pattern in which a background thread generates new plot data, and a UI thread (e.g., in
        a Jupyter notebook) periodically calls `update_plot` to redraw the plot with the latest data.

        :param render: Whether to render the plot data. If False, then plot data will be updated but nothing will
        be shown.
        :param pdf: PDF to plot to, or None to show directly.
        :return: Matplotlib figure, if one was generated and not plotting to PDF.
        """

1	from abc import ABC	×
2	from typing import Optional	×
3
4	from matplotlib import pyplot as plt	×
5	from matplotlib.backends.backend_pdf import PdfPages	×
6
7	from rlai.models import FunctionApproximationModel	×
8
9
10	class StateFunctionApproximationModel(FunctionApproximationModel, ABC):	×
11	"""
12	Base class for models that approximate state-action value functions.
13	"""
14
15	def plot(	×
16	self,
17	render: bool,
18	pdf: Optional[PdfPages]
19	) -> Optional[plt.Figure]:
20	"""
21	Plot the model. If called from the main thread and render is True, then a new plot will be generated. If called
22	from a background thread, then the data used by the plot will be updated but a plot will not be generated or
23	updated. This supports a pattern in which a background thread generates new plot data, and a UI thread (e.g., in
24	a Jupyter notebook) periodically calls `update_plot` to redraw the plot with the latest data.
25
26	:param render: Whether to render the plot data. If False, then plot data will be updated but nothing will
27	be shown.
28	:param pdf: PDF to plot to, or None to show directly.
29	:return: Matplotlib figure, if one was generated and not plotting to PDF.
30	"""

MatthewGerber / rlai / 27471184649

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous