Skip to content

Commit bcf044c

Browse files
committed
updated tests
1 parent f52f80b commit bcf044c

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

src/lib.rs

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ pub mod translators;
99

1010
#[cfg(test)]
1111
mod tests {
12+
#[cfg(not(feature = "offline_req"))]
13+
use std::collections::HashMap;
1214
use dotenv::dotenv;
1315
#[cfg(feature = "offline_req")]
1416
use model_manager::model_manager::ModelManager;
@@ -20,17 +22,29 @@ mod tests {
2022
use crate::languages::Language;
2123
#[cfg(feature = "offline_req")]
2224
use crate::model_register::register;
25+
#[cfg(not(feature = "offline_req"))]
26+
use crate::translators::chainer::TranslatorInfo;
27+
#[cfg(not(feature = "offline_req"))]
28+
use crate::translators::chainer::TranslatorSelectorInfo;
2329
use crate::translators::dev::{get_csv_errors, get_languages};
30+
#[cfg(feature = "offline_req")]
2431
use crate::translators::offline::ctranslate2::model_management::{
2532
CTranslateModels, ModelLifetime, TokenizerModels,
2633
};
34+
#[cfg(feature = "offline_req")]
2735
use crate::translators::offline::ctranslate2::Device;
36+
#[cfg(feature = "offline_req")]
2837
use crate::translators::offline::m2m100::{M2M100ModelType, M2M100Translator};
38+
#[cfg(feature = "offline_req")]
2939
use crate::translators::offline::ModelFormat;
3040
use crate::translators::scrape::papago::PapagoTranslator;
3141
use crate::translators::tokens::Tokens;
32-
use crate::translators::translator_structure::{TranslatorCTranslate, TranslatorLanguages};
33-
use crate::translators::Translator;
42+
#[cfg(feature = "offline_req")]
43+
use crate::translators::translator_structure::TranslatorCTranslate;
44+
use crate::translators::translator_structure::TranslatorLanguages;
45+
use crate::translators::{Translator, Translators};
46+
#[cfg(not(feature = "offline_req"))]
47+
use crate::translators::context::Context;
3448

3549
#[tokio::test]
3650
#[cfg(feature = "offline_req")]

0 commit comments

Comments
 (0)