12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
python/gtsam/examples/SFMExample_bal.py
View File

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