FIX: Log a warning instead of raising an AssertionError

[ketangangal]

Describe changes

I fixed issue : Log a warning instead of raising an AssertionError

else:
    # TODO [ENG-895]: Refactor this to raise a warning instead.
    raise AssertionError(
        "Cannot set magic flag in non-notebook environments."
    )

to

else:
    cli_utils.warning(
        "Cannot set magic flag in non-notebook environments."
    )

[strickvl]

Thanks for these suggested fixes, @ketangangal!

Some small changes still on this, since the imports weren't quite correct.

[strickvl]

@ketangangal just wanted to let you know that you can just update your PR from within this one. No real need to create a new one. It helps keep all the work that was done in one place. We can then see how things developed as a result of feedback etc.

[ketangangal]

@strickvl got it thanks correcting mistakes.

[strickvl]

One other thing I just remembered from #616 is that the imports should probably be as in my suggested change (i.e. cli_utils.warning and the import change as suggested).

The way you did it before was technically correct, but it just didn't fit how we do it usually in the codebase.

[ketangangal]

@strickvl done !

[strickvl]

@strickvl done !

@ketangangal You can see in the CI testing that we have a bunch of failures there. Looking at the error for one of those, it says that the imports are unsorted:

ERROR: /Users/runner/work/zenml/zenml/src/zenml/integrations/dash/visualizers/pipeline_run_lineage_visualizer.py Imports are incorrectly sorted and/or formatted.

I think you didn't run the Poetry linting script as suggested in the CONTRIBUTING.md file. I'd suggest you do that and resubmit. Then we can make sure it's really passing our tests.

[strickvl]

@ketangangal I took the liberty of running the linting script on your behalf. If the tests pass I'll be sure to merge this in. Thank you for your contribution!