linting issues

Author: Andrei Bratu

tests/utilities/test_prompt.py

@@ -361,7 +361,7 @@ def call_llm_with_hl_call(messages: list[dict]):
     assert len(response.items) == 1  # type: ignore
 
 
-@pytest.mark.skip(retries=3, delay=5)
+@pytest.mark.flaky(retries=3, delay=5)
 def test_overridden_call_with_prompt_in_prompt(
     humanloop_client: Humanloop,
     test_directory: DirectoryIdentifiers,
@@ -410,7 +410,7 @@ def outer_call_llm_with_hl_call(messages: list[dict]):
     )
 
     # Wait for the workspace to be updated
-    time.wait(3)
+    time.sleep(3)
 
     assert output is not None
     response = humanloop_client.directories.get(id=test_directory.id)