Export el envs

Author: ishii-norimi

create_import_list.js

@@ -1,16 +1,12 @@
 import fs from 'fs'
 
-const modelFiles = await fs.promises.readdir('./lib/model')
-
 let code = `import { Tree, Tensor, Matrix } from './util/math.js'
 import EnsembleBinaryModel from './util/ensemble.js'
 
 `
 
-const wholename = []
-
-for (const modelName of modelFiles) {
-	const mod = await import(`./lib/model/${modelName}`)
+const createImportStatement = async name => {
+	const mod = await import(`./lib/${name}`)
 	let d = null
 	const named = []
 	for (const name of Object.keys(mod)) {
@@ -21,16 +17,32 @@ for (const modelName of modelFiles) {
 		}
 	}
 	if (d) {
-		wholename.push(d.name)
 		if (named.length === 0) {
-			code += `import ${d.name} from './model/${modelName}'\n`
+			code += `import ${d.name} from './${name}'\n`
 		} else {
-			code += `import ${d.name}, { ${named.join(', ')} } from './model/${modelName}'\n`
+			code += `import ${d.name}, { ${named.join(', ')} } from './${name}'\n`
 		}
+		named.push(d.name)
 	} else if (named.length > 0) {
-		code += `import { ${named.join(', ')} } from './model/${modelName}'\n`
+		code += `import { ${named.join(', ')} } from './${name}'\n`
 	}
-	wholename.push(...named)
+	return named
+}
+
+const modelFiles = await fs.promises.readdir('./lib/model')
+const modelNames = []
+
+for (const modelName of modelFiles) {
+	const named = await createImportStatement(`model/${modelName}`)
+	modelNames.push(...named)
+}
+
+const rlFiles = await fs.promises.readdir('./lib/rl')
+const rlNames = []
+
+for (const rlName of rlFiles) {
+	const named = await createImportStatement(`rl/${rlName}`)
+	rlNames.push(...named)
 }
 
 code += `
@@ -41,12 +53,19 @@ export default {
 	models: {
 `
 
-for (const name of wholename) {
+for (const name of modelNames) {
 	code += `		${name},\n`
 }
 code += `
 	},
 	EnsembleBinaryModel,
+	rl: {
+`
+for (const name of rlNames) {
+	code += `		${name},\n`
+}
+code += `
+	}
 }
 `