<commit_before># cython: boundscheck=False
# cython: nonecheck=False
# cython: cdivision=True
# cython: wraparound=False
# cython: profile=False
# cython: language_level=3

# Third-party
from astropy.utils.misc import isiterable
import cython
import astropy.units as u
import numpy as np
cimport numpy as np

# This package
from .. import combine, Orbit
from ..nbody import DirectNBody
from ...potential import Hamiltonian, PotentialBase, StaticFrame
from ...potential.potential.cpotential cimport CPotentialWrapper, CPotential
from ...potential.hamiltonian.chamiltonian import Hamiltonian

from ._coord cimport cross, norm, apply_3matrix

cdef extern from "potential/src/cpotential.h":
    double c_d2_dr2(CPotential *p, double t, double *q, double *epsilon) nogil


cdef class BaseStreamDF:

    @cython.embedsignature(True)
    def __init__(self, lead=True, trail=True, random_state=None):
        """TODO: documentation"""<commit_msg>docstring<commit_after># cython: boundscheck=False
# cython: nonecheck=False
# cython: cdivision=True
# cython: wraparound=False
# cython: profile=False
# cython: language_level=3

# Third-party
from astropy.utils.misc import isiterable
import cython
import astropy.units as u
import numpy as np
cimport numpy as np

# This package
from .. import combine, Orbit
from ..nbody import DirectNBody
from ...potential import Hamiltonian, PotentialBase, StaticFrame
from ...potential.potential.cpotential cimport CPotentialWrapper, CPotential
from ...potential.hamiltonian.chamiltonian import Hamiltonian

from ._coord cimport cross, norm, apply_3matrix

cdef extern from "potential/src/cpotential.h":
    double c_d2_dr2(CPotential *p, double t, double *q, double *epsilon) nogil


cdef class BaseStreamDF:

    @cython.embedsignature(True)
    def __init__(self, lead=True, trail=True, random_state=None):
        """A distribution function for stellar streams.

        This class specifies how massless star particles should be sampled in
        order to generate a mock stellar stream.

        Parameters
        ----------
        lead : bool (optional)
            Generate a leading tail. Default: True.
        trail : bool (optional)
            Generate a trailing tail. Default: True.
        random_state : `~numpy.random.RandomState` (optional)
            To control random number generation.

        """
{'fraction_non_alphanumeric': 0.08669783255418614, 'fraction_numerical': 0.003399915002124947, 'mean_word_length': 3.9767441860465116, 'pattern_counts': {'":': 2, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '102015', 'n_tokens_mistral': 697, 'n_tokens_neox': 673, 'n_words': 224}