Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Test and handle all tensor dtypes as images #1840

Merged
merged 12 commits into from
Apr 13, 2023
Merged

Test and handle all tensor dtypes as images #1840

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
dcfbc1a
Support i64 and u64 tensors
emilk Apr 13, 2023
daada52
api_demo: log all image types
emilk Apr 13, 2023
01f5def
Don't even print out the contents of a tensor
emilk Apr 13, 2023
573cbb8
Handle unfilterable float textures
emilk Apr 13, 2023
dbb4c73
fix typo
emilk Apr 13, 2023
1f02dac
py-format
emilk Apr 13, 2023
54d99b8
Simplify is_float_filterable
emilk Apr 13, 2023
128b895
Add a helper function pad_and_narrow_and_cast
emilk Apr 13, 2023
067a4a9
Reuse existing image
emilk Apr 13, 2023
9fe81a8
Exclude image_tensors demo from default api_demo
emilk Apr 13, 2023
9fe5ff0
Still run all api demos in e2e test
emilk Apr 13, 2023
f973e5f
pyformat
emilk Apr 13, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 7 additions & 3 deletions examples/python/api_demo/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -329,17 +329,21 @@ def main() -> None:

parser = argparse.ArgumentParser(description="Logs rich data using the Rerun SDK.")
parser.add_argument(
"--demo", type=str, default="all", help="What demo to run", choices=["all"] + list(demos.keys())
"--demo", type=str, default="most", help="What demo to run", choices=["most", "all"] + list(demos.keys())
)

rr.script_add_args(parser)
args = parser.parse_args()

rr.script_setup(args, "api_demo")

if args.demo == "all":
print("Running all demos…")
if args.demo in ["most", "all"]:
print(f"Running {args.demo} demos…")
for name, demo in demos.items():
# Some demos are just a bit… too much
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

😆 👍

if args.demo == "most" and name in ["image_tensors"]:
continue

logging.info(f"Starting {name}")
demo()
else:
Expand Down