Skip to content

Commit 5304576

Browse files
committed
fix: test cases
1 parent eb94a75 commit 5304576

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tests/test_async_moderations.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,9 @@ def get_metadata(self):
4747
t1_params = []
4848
t = []
4949
for k, v in models.items():
50-
for i in v["chat"]:
51-
t.append((client, k, os.environ.get(v["env_variable"]), i))
50+
if k == "openai":
51+
for i in v["chat"]:
52+
t.append((client, k, os.environ.get(v["env_variable"]), i))
5253

5354
t1_params.extend(t)
5455

tests/test_moderations.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,9 @@ def get_metadata(self):
4747
t1_params = []
4848
t = []
4949
for k, v in models.items():
50-
for i in v["chat"]:
51-
t.append((client, k, os.environ.get(v["env_variable"]), i))
50+
if k == "openai":
51+
for i in v["chat"]:
52+
t.append((client, k, os.environ.get(v["env_variable"]), i))
5253

5354
t1_params.extend(t)
5455

0 commit comments

Comments
 (0)