Fix formatting error in src/pipelines/masked_language.rs

src/lib.rs

@@ -360,15 +360,15 @@
 //! # use rust_bert::pipelines::zero_shot_classification::ZeroShotClassificationModel;
 //! # fn main() -> anyhow::Result<()> {
 //! let sequence_classification_model = ZeroShotClassificationModel::new(Default::default())?;
-//!  let input_sentence = "Who are you voting for in 2020?";
-//!  let input_sequence_2 = "The prime minister has announced a stimulus package which was widely criticized by the opposition.";
-//!  let candidate_labels = &["politics", "public health", "economics", "sports"];
-//!  let output = sequence_classification_model.predict_multilabel(
-//!      &[input_sentence, input_sequence_2],
-//!      candidate_labels,
-//!      None,
-//!      128,
-//!  );
+//! let input_sentence = "Who are you voting for in 2020?";
+//! let input_sequence_2 = "The prime minister has announced a stimulus package which was widely criticized by the opposition.";
+//! let candidate_labels = &["politics", "public health", "economics", "sports"];
+//! let output = sequence_classification_model.predict_multilabel(
+//!     &[input_sentence, input_sequence_2],
+//!     candidate_labels,
+//!     None,
+//!     128,
+//! );
 //! # Ok(())
 //! # }
 //! ```

src/pipelines/masked_language.rs

@@ -15,15 +15,15 @@
 //! a masked word can be specified in the `MaskedLanguageConfig` (`mask_token`). and allows
 //! multiple masked tokens per input sequence.
 //!
-//!  ```no_run
-//!use rust_bert::bert::{BertConfigResources, BertModelResources, BertVocabResources};
-//!use rust_bert::pipelines::common::ModelType;
-//!use rust_bert::pipelines::masked_language::{MaskedLanguageConfig, MaskedLanguageModel};
-//!use rust_bert::resources::RemoteResource;
-//! fn main() -> anyhow::Result<()> {
+//! ```no_run
+//! use rust_bert::bert::{BertConfigResources, BertModelResources, BertVocabResources};
+//! use rust_bert::pipelines::common::ModelType;
+//! use rust_bert::pipelines::masked_language::{MaskedLanguageConfig, MaskedLanguageModel};
+//! use rust_bert::resources::RemoteResource;
 //!
+//! fn main() -> anyhow::Result<()> {
 //!     use rust_bert::pipelines::common::ModelResource;
-//! let config = MaskedLanguageConfig::new(
+//!     let config = MaskedLanguageConfig::new(
 //!         ModelType::Bert,
 //!         ModelResource::Torch(Box::new(RemoteResource::from_pretrained(BertModelResources::BERT))),
 //!         RemoteResource::from_pretrained(BertConfigResources::BERT),

src/pipelines/mod.rs

@@ -205,15 +205,15 @@
 //! # use rust_bert::pipelines::zero_shot_classification::ZeroShotClassificationModel;
 //! # fn main() -> anyhow::Result<()> {
 //! let sequence_classification_model = ZeroShotClassificationModel::new(Default::default())?;
-//!  let input_sentence = "Who are you voting for in 2020?";
-//!  let input_sequence_2 = "The prime minister has announced a stimulus package which was widely criticized by the opposition.";
-//!  let candidate_labels = &["politics", "public health", "economics", "sports"];
-//!  let output = sequence_classification_model.predict_multilabel(
-//!      &[input_sentence, input_sequence_2],
-//!      candidate_labels,
-//!      None,
-//!      128,
-//!  );
+//! let input_sentence = "Who are you voting for in 2020?";
+//! let input_sequence_2 = "The prime minister has announced a stimulus package which was widely criticized by the opposition.";
+//! let candidate_labels = &["politics", "public health", "economics", "sports"];
+//! let output = sequence_classification_model.predict_multilabel(
+//!     &[input_sentence, input_sequence_2],
+//!     candidate_labels,
+//!     None,
+//!     128,
+//! );
 //! # Ok(())
 //! # }
 //! ```

src/pipelines/zero_shot_classification.rs

@@ -25,15 +25,15 @@
 //! # use rust_bert::pipelines::zero_shot_classification::ZeroShotClassificationModel;
 //! # fn main() -> anyhow::Result<()> {
 //! let sequence_classification_model = ZeroShotClassificationModel::new(Default::default())?;
-//!  let input_sentence = "Who are you voting for in 2020?";
-//!  let input_sequence_2 = "The prime minister has announced a stimulus package which was widely criticized by the opposition.";
-//!  let candidate_labels = &["politics", "public health", "economics", "sports"];
-//!  let output = sequence_classification_model.predict_multilabel(
-//!      &[input_sentence, input_sequence_2],
-//!      candidate_labels,
-//!      None,
-//!      128,
-//!  );
+//! let input_sentence = "Who are you voting for in 2020?";
+//! let input_sequence_2 = "The prime minister has announced a stimulus package which was widely criticized by the opposition.";
+//! let candidate_labels = &["politics", "public health", "economics", "sports"];
+//! let output = sequence_classification_model.predict_multilabel(
+//!     &[input_sentence, input_sequence_2],
+//!     candidate_labels,
+//!     None,
+//!     128,
+//! );
 //! # Ok(())
 //! # }
 //! ```