Skip to content

Commit

Permalink
Ignore the warning from the logging_timer which happens whenever DFPS…
Browse files Browse the repository at this point in the history
…plitUsersStage doesn't have any output messages
  • Loading branch information
dagardner-nv committed Jan 9, 2024
1 parent 21d8562 commit 03168a2
Showing 1 changed file with 7 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
import json
import os
import typing
import warnings

import pytest

Expand Down Expand Up @@ -99,7 +100,12 @@ def test_extract_users(config: Config,
skip_users=skip_users,
only_users=only_users)

results = stage.extract_users(df)
with warnings.catch_warnings():
# Ignore warning about the log message not being set. This happens whenever there aren't any output_messages
warnings.filterwarnings("ignore",
message="Must set log msg before end of context! Skipping log",
category=UserWarning)
results = stage.extract_users(df)

if not include_generic and not include_individual:
# Extra check for weird combination
Expand Down

0 comments on commit 03168a2

Please sign in to comment.