add return type definitions

Because my time is more valuable than a reviewer's pedanticness
release/4.3a0
Varun Agrawal 2021-11-10 14:27:28 -05:00
parent 5c9c60a0be
commit 0ccb18b055
1 changed files with 5 additions and 5 deletions

View File

@ -17,8 +17,8 @@ import sys
import gtsam
from gtsam import (GeneralSFMFactorCal3Bundler,
PriorFactorPinholeCameraCal3Bundler, PriorFactorPoint3)
from gtsam.symbol_shorthand import C, P
from gtsam.utils import plot
from gtsam.symbol_shorthand import C, P # type: ignore
from gtsam.utils import plot # type: ignore
from matplotlib import pyplot as plt
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
@ -26,7 +26,7 @@ logging.basicConfig(stream=sys.stdout, level=logging.INFO)
DEFAULT_BAL_DATASET = "dubrovnik-3-7-pre"
def plot_scene(scene_data: gtsam.SfmData, result: gtsam.Values):
def plot_scene(scene_data: gtsam.SfmData, result: gtsam.Values) -> None:
"""Plot the SFM results."""
plot_vals = gtsam.Values()
for cam_idx in range(scene_data.number_cameras()):
@ -41,7 +41,7 @@ def plot_scene(scene_data: gtsam.SfmData, result: gtsam.Values):
plt.show()
def run(args: argparse.Namespace):
def run(args: argparse.Namespace) -> None:
""" Run LM optimization with BAL input data and report resulting error """
input_file = args.input_file
@ -109,7 +109,7 @@ def run(args: argparse.Namespace):
plot_scene(scene_data, result)
def main():
def main() -> None:
"""Main runner."""
parser = argparse.ArgumentParser()
parser.add_argument('-i',