diff --git a/aeon/dj_pipeline/utils/plotting.py b/aeon/dj_pipeline/utils/plotting.py index 4704b221..5a160b70 100644 --- a/aeon/dj_pipeline/utils/plotting.py +++ b/aeon/dj_pipeline/utils/plotting.py @@ -25,16 +25,17 @@ def plot_reward_rate_differences(subject_keys): - """Plotting the reward rate differences between food patches (Patch 2 - Patch 1) for all sessions from all subjects specified in "subject_keys". + """Plots the reward rate differences between two food patches (Patch 2 - Patch 1). - Examples: - ``` - subject_keys = - (acquisition.Experiment.Subject & 'experiment_name = "exp0.1-r0"').fetch('KEY') + The reward rate differences between the two food patches are plotted + for all sessions from all subjects in ``subject_keys``. - fig = plot_reward_rate_differences(subject_keys) - ``` - """ # noqa E501 + Examples: + >>> subject_keys = ( + ... acquisition.Experiment.Subject + ... & 'experiment_name = "exp0.1-r0"').fetch('KEY') + >>> fig = plot_reward_rate_differences(subject_keys) + """ subj_names, sess_starts, rate_timestamps, rate_diffs = ( analysis.InArenaRewardRate & subject_keys ).fetch("subject", "in_arena_start", "pellet_rate_timestamps", "patch2_patch1_rate_diff")