1 files changed, 4 insertions, 0 deletions
diff --git a/cli/factory.rs b/cli/factory.rs
index bd457e2ab..4e7138938 100644
--- a/cli/factory.rs
+++ b/cli/factory.rs
@@ -668,6 +668,10 @@ impl CliFactory {
   ) -> Result<CliMainWorkerOptions, AnyError> {
     Ok(CliMainWorkerOptions {
       argv: self.options.argv().clone(),
+      // This optimization is only available for "run" subcommand
+      // because we need to register new ops for testing and jupyter
+      // integration.
+      skip_op_registration: self.options.sub_command().is_run(),
       log_level: self.options.log_level().unwrap_or(log::Level::Info).into(),
       coverage_dir: self.options.coverage_dir(),
       enable_op_summary_metrics: self.options.enable_op_summary_metrics(),