Add return checker, needed 4 fixes

This commit is contained in:
Julian Stirling 2025-06-10 14:00:56 +01:00
parent 3f0564f253
commit 2050c2dc7c
5 changed files with 14 additions and 23 deletions

View file

@ -108,7 +108,7 @@ select = [
# "LOG", # Flake8 logging issues (pedantic logger formatting issues can be added with "G") # "LOG", # Flake8 logging issues (pedantic logger formatting issues can be added with "G")
# "T20", # Warns for print statments, production code should log # "T20", # Warns for print statments, production code should log
# "PT", # pytest linting # "PT", # pytest linting
# "RET", # Consistent clear return statments "RET", # Consistent clear return statments
"RSE", # Raise parentheses "RSE", # Raise parentheses
# "SIM", # Simplifications detected # "SIM", # Simplifications detected
"ARG", # unused arguments "ARG", # unused arguments

View file

@ -118,20 +118,12 @@ class RecentringThing(Thing):
logging.info( logging.info(
f"Breaking because the highest point is at {np.argmax(sorted_all_heights)} in the list" f"Breaking because the highest point is at {np.argmax(sorted_all_heights)} in the list"
) )
# plt.plot(sorted_lateral, sorted_all_heights,'.')
# plt.plot(sorted_lateral, quad_fit_func(sorted_lateral))
# plt.show()
break break
else: if turning_loc < np.min(sorted_lateral):
if turning_loc < np.min(sorted_lateral): moves = -1
moves = -1 elif turning_loc > np.max(sorted_lateral):
elif turning_loc > np.max(sorted_lateral): moves = 1
moves = 1
else:
# plt.plot(sorted_lateral, sorted_all_heights,'.')
# plt.plot(sorted_lateral, quad_fit_func(sorted_lateral))
# plt.show()
pass
# Centre value is replaced by the maximum value recorded in that axis # Centre value is replaced by the maximum value recorded in that axis
centre[direction] = focused_pos[direction][np.argmax(all_heights)][ centre[direction] = focused_pos[direction][np.argmax(all_heights)][

View file

@ -105,11 +105,12 @@ class JPEGSharpnessMonitor:
stop: int = int(np.argmax(jpeg_times > stage_times[1])) stop: int = int(np.argmax(jpeg_times > stage_times[1]))
except ValueError as e: except ValueError as e:
if np.sum(jpeg_times > stage_times[0]) == 0: if np.sum(jpeg_times > stage_times[0]) == 0:
raise ValueError( errmsg = (
"No images were captured during the move of the stage. Perhaps the camera is not streaming images?" "No images were captured during the move of the stage. "
) from e "Perhaps the camera is not streaming images?"
else: )
raise e raise ValueError(errmsg) from e
raise e
if stop < 1: if stop < 1:
stop = len(jpeg_times) stop = len(jpeg_times)
logging.debug("changing stop to %s", (stop)) logging.debug("changing stop to %s", (stop))

View file

@ -23,8 +23,7 @@ class BackgroundDetectThing(Thing):
bd = self.thing_settings.get("background_distributions", None) bd = self.thing_settings.get("background_distributions", None)
if bd: if bd:
return ChannelDistributions(**bd) return ChannelDistributions(**bd)
else: return None
return None
@background_distributions.setter @background_distributions.setter
def background_distributions(self, value: Optional[ChannelDistributions]) -> None: def background_distributions(self, value: Optional[ChannelDistributions]) -> None:

View file

@ -189,8 +189,7 @@ def get_expected_result_for_example_smart_spiral(
""" """
pkl_fname = os.path.join(THIS_DIR, f"example_smart_spiral_{sample_name}.pkl") pkl_fname = os.path.join(THIS_DIR, f"example_smart_spiral_{sample_name}.pkl")
with open(pkl_fname, "rb") as pkl_file_obj: with open(pkl_fname, "rb") as pkl_file_obj:
planner = pickle.load(pkl_file_obj) return pickle.load(pkl_file_obj)
return planner
def load_sample_points(sample_name: str): def load_sample_points(sample_name: str):