diff --git a/packages/opentelemetry-instrumentation-replicate/tests/test_hello.py b/packages/opentelemetry-instrumentation-replicate/tests/test_hello.py deleted file mode 100644 index 56970305c..000000000 --- a/packages/opentelemetry-instrumentation-replicate/tests/test_hello.py +++ /dev/null @@ -1,8 +0,0 @@ -"""Hello unit test module.""" - -from opentelemetry.instrumentation.replicate.hello import hello - - -def test_hello(): - """Test the hello function.""" - assert hello() == "Hello opentelemetry-instrumentation-replicate" diff --git a/packages/traceloop-sdk/tests/test_replicate_instrumentaion.py b/packages/traceloop-sdk/tests/test_replicate_instrumentaion.py deleted file mode 100644 index 86681cb29..000000000 --- a/packages/traceloop-sdk/tests/test_replicate_instrumentaion.py +++ /dev/null @@ -1,73 +0,0 @@ -import replicate -from traceloop.sdk.decorators import workflow, task - - -def test_replicate_image_generation(exporter): - @task(name="image_generation") - def generate_image(): - replicate.run( - "stability-ai/stable-diffusion:27b93a2413e7f36cd83da926f3656280b2931564ff050bf9575f1fdf9bcd7478", - input={"prompt": "robots"}, - ) - - @workflow(name="robot_image_generator") - def image_workflow(): - generate_image() - - image_workflow() - - spans = exporter.get_finished_spans() - assert [span.name for span in spans] == [ - "replicate.run", - "image_generation.task", - "robot_image_generator.workflow", - ] - - -def test_replicate_image_generation_stream(exporter): - @task(name="image_generation_stream") - def generate_image_stream(): - model_version = "meta/llama-2-70b-chat:02e509c789964a7ea8736978a43525956ef40397be9033abf9fd2badfe68c9e3" - for event in replicate.stream( - model_version, - input={ - "prompt": "robots", - }, - ): - continue - - @workflow(name="robot_image_generator_stream") - def image_workflow(): - generate_image_stream() - - image_workflow() - - spans = exporter.get_finished_spans() - assert [span.name for span in spans] == [ - "replicate.stream", - "image_generation_stream.task", - "robot_image_generator_stream.workflow", - ] - - -def test_replicate_image_generation_predictions(exporter): - @task(name="image_generation_predictions") - def generate_image_predictions(): - model = replicate.models.get("kvfrans/clipdraw") - version = model.versions.get( - "5797a99edc939ea0e9242d5e8c9cb3bc7d125b1eac21bda852e5cb79ede2cd9b" - ) - replicate.predictions.create(version, input={"prompt": "robots"}) - - @workflow(name="robot_image_generator_predictions") - def image_workflow(): - generate_image_predictions() - - image_workflow() - - spans = exporter.get_finished_spans() - assert [span.name for span in spans] == [ - "replicate.predictions.create", - "image_generation_predictions.task", - "robot_image_generator_predictions.workflow", - ]