From 287a5fc8da234c05deecd71c4cdc84323e01598e Mon Sep 17 00:00:00 2001 From: Eric Neilsen Date: Thu, 2 May 2024 10:11:34 -0700 Subject: [PATCH] be compatible with surveys for which ra_deg is not iterable --- schedview/plot/survey.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/schedview/plot/survey.py b/schedview/plot/survey.py index f97df0e8..154324d6 100644 --- a/schedview/plot/survey.py +++ b/schedview/plot/survey.py @@ -203,12 +203,19 @@ def map_survey_healpix( if survey is not None: try: + try: + ra_deg = list(survey.ra_deg) + dec_deg = list(survey.dec_deg) + except TypeError: + ra_deg = [survey.ra_deg] + dec_deg = [survey.dec_deg] + survey_field_data_source = bokeh.models.ColumnDataSource( data={ - "ra": list(survey.ra_deg), - "decl": list(survey.dec_deg), - "name": ["survey_pointing {i}" for i, ra in enumerate(list(survey.ra_deg))], - "glyph_size": [20] * len(list(survey.ra_deg)), + "ra": ra_deg, + "decl": dec_deg, + "name": ["survey_pointing {i}" for i, ra in enumerate(ra_deg)], + "glyph_size": [20] * len(ra_deg), }, name="survey_pointings", )